莫煩Pytorch系列教程2 3 激勵函式

2021-10-06 10:44:56 字數 1535 閱讀 9999

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...