Pytorch中的forward有哪些功能

今天小編給大家分享一下Pytorch中的forward有哪些功能的相關知識點，內(nèi)容詳細，邏輯清晰，相信大部分人都還太了解這方面的知識，所以分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后有所收獲，下面我們一起來了解一下吧。

forward有什么特殊功能？

第一條：.forward()可以不寫

我最開始發(fā)現(xiàn)forward()的與眾不同之處就是在此，首先舉個例子：

import torch.nn as nn
class test(nn.Module):
    def __init__(self, input):
        super(test,self).__init__()
        self.input = input

    def forward(self,x):
        return self.input * x

T = test(8)
print(T(6))

# print(T.forward(6))
--------------------------運行結果-------------------------
D:\Users\Lenovo\anaconda3\python.exe C:/Users/Lenovo/Desktop/DL/pythonProject/tt.py
48

Process finished with exit code 0

可以發(fā)現(xiàn)，T(6)是可以輸出的！而且不用指定，默認了調用forward方法。當然如果非要寫上.forward()這也是可以正常運行的，和不寫是一樣的。

如果不調用Pytorch(正常的Python語法規(guī)則)，這樣肯定會報錯的

# import torch.nn as nn  #不再調用torch
class test():
    def __init__(self, input):
        self.input = input

    def forward(self,x):
        return self.input * x

T = test(8)
print(T.forward(6))
print("************************")
print(T(6))
--------------------------運行結果-------------------------
D:\Users\Lenovo\anaconda3\python.exe C:/Users/Lenovo/Desktop/DL/pythonProject/tt.py
48
************************
Traceback (most recent call last):
  File "C:\Users\Lenovo\Desktop\DL\pythonProject\tt.py", line 77, in <module>
    print(T(6))
TypeError: 'test' object is not callable

Process finished with exit code 1

這里會報：‘test’ object is not callable
因為class不能被直接調用，不知道你想調用哪個方法。

第二條：優(yōu)先運行forward方法

如果在class中再增加一個方法：

import torch.nn as nn
class test(nn.Module):
    def __init__(self, input):
        super(test,self).__init__()
        self.input = input

    def byten(self):
        return self.input * 10

    def forward(self,x):
        return self.input * x

T = test(8)
print(T(6))
print(T.byten())
--------------------------運行結果-------------------------
D:\Users\Lenovo\anaconda3\python.exe C:/Users/Lenovo/Desktop/DL/pythonProject/tt.py
48
80

Process finished with exit code 0

可以見到，在class中有多個method的時候，如果不指定method，forward是會被優(yōu)先執(zhí)行的。

以上就是“Pytorch中的forward有哪些功能”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家閱讀完這篇文章都有很大的收獲，小編每天都會為大家更新不同的知識，如果還想學習更多的知識，請關注億速云行業(yè)資訊頻道。