import torch
import torch.nn.functional as f #激勵函式都在這裡
from torch.autograd import variable
#做一些假資料
x = torch.linspace(-5
,5,200
)#在-5到5的區間裡,取200個點
print
(x.size())
print
(x.shape)
print
(x.ndim)
x= variable(x)
x_np = x.data.numpy(
)#換成numpy,出圖時使用
#幾種常見的激勵函式
y_relu = f.relu(x)
.data.numpy(
)y_simoid = f.sigmoid(x)
.data.numpy(
)y_tanh = f.tanh(x)
.data.numpy(
)y_softplus = f.softplus(x)
.data.numpy(
)#畫圖教程
import matplotlib.pyplot as plt
plt.figure(
1,figsize=(8
,6))
plt.subplot(
221)
plt.plot(x_np,y_relu,c=
'red'
,label =
'relu'
)plt.ylim((-
1,5)
)plt.legend(loc =
'best'
)plt.subplot(
222)
plt.plot(x_np,y_simoid,c=
'red'
,label =
'relu'
)plt.ylim((-
0.2,
1.2)
)plt.legend(loc =
'best'
)#顯示圖例
莫煩tensorflow系列教程學習
1.普通機器學習 函式係數 y 0.1x 0.3 coding gbk import tensorflow as tf import numpy as np 生成資料,y 0.1x 0.3 x data np.random rand 100 astype np.float32 y data x da...
莫煩pytorch學習筆記
此處x,y為資料集的tensor torch dataset data.tensordataset data tensor x,target tensor y loader data.dataloader dataset torch dataset,batch size batch size,shu...
莫煩pytorch批訓練
import torch import torch.utils.data as data 包裝資料類 tensordataset 包裝資料和目標張量的資料集,通過沿著第乙個維度索引兩個張量來 class torch.utils.data.tensordataset data tensor,targe...