Pytorch深度学习-神经网络的基本骨架-nn.Moudle的使用（neural network）（小土堆）

Mr chenxizhi

于 2024-07-08 18:03:42 发布

阅读量224

点赞数 1

文章标签：深度学习 pytorch 神经网络

本文链接：https://blog.csdn.net/weixin_53141706/article/details/140275658

版权

containers.Module

查阅网址：https://pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module
过程

input->forward()->output

代码示例

import torch  
import torch.nn as nn  
  
class nn_test(nn.Module):
	#重写方法
    def __init__(self):  
        super().__init__()  
  
    def forward(self,input):  
        output = input + 1  
        return output  

#实例化类
nnTest = nn_test()  
x = torch.tensor(1.0)  
output = nnTest(x)  
print(output)