溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

pytorch如何激活函數ReLU

發布時間:2021-08-17 11:29:52 來源:億速云 閱讀:169 作者:小新 欄目:開發技術

這篇文章將為大家詳細講解有關pytorch如何激活函數ReLU,小編覺得挺實用的,因此分享給大家做個參考,希望大家閱讀完這篇文章后可以有所收獲。

測試代碼:

import torch
import torch.nn as nn

#inplace為True,將會改變輸入的數據 ,否則不會改變原輸入,只會產生新的輸出
m = nn.ReLU(inplace=True)
input = torch.randn(7)

print("輸入處理前圖片:")
print(input)

output = m(input)

print("ReLU輸出:")
print(output)
print("輸出的尺度:")
print(output.size())

print("輸入處理后圖片:")
print(input)

輸出為:

輸入處理前圖片:

tensor([ 1.4940, 1.0278, -1.9883, -0.1871, 0.4612, 0.0297, 2.4300])

ReLU輸出:

tensor([ 1.4940, 1.0278, 0.0000, 0.0000, 0.4612, 0.0297, 2.4300])

輸出的尺度:

torch.Size([7])

輸入處理后圖片:

tensor([ 1.4940, 1.0278, 0.0000, 0.0000, 0.4612, 0.0297, 2.4300])

結論:

nn.ReLU(inplace=True)

inplace為True,將會改變輸入的數據 ,否則不會改變原輸入,只會產生新的輸出

關于“pytorch如何激活函數ReLU”這篇文章就分享到這里了,希望以上內容可以對大家有一定的幫助,使各位可以學到更多知識,如果覺得文章不錯,請把它分享出去讓更多的人看到。

向AI問一下細節

免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。

AI

亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女