神经网络的基本骨架--nn,module的使用
- 官网Module介绍
- Python父类子类继承关系
- 前向神经网络
- pycharm快捷键重写类方法
- code
- debug
B站小土堆视频学习笔记
官网Module介绍
https://pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module
Python父类子类继承关系
https://www.php.cn/faq/555151.html
前向神经网络
前向传播
与之对应的还有反向传播
推荐课程
吴恩达卷积
李宏毅深度学习 cnn
同济子豪兄神经网络
pycharm快捷键重写类方法
code–>Generate–>override Methods–>init 重写init方法
code
import torch
from torch import nn
class my_neural_network(nn.Module):
def __init__(self):
super().__init__()
def forward(self, input):
output = input + 1
return output
my_net = my_neural_network()
# x = torch.Tensor(1.0) # TypeError: new(): data must be a sequence (got float)
x = torch.tensor(1.0)
output = my_net(x)
print(output)
debug
torch.tensor和torch.Tensor的区别
torch.tensor是方法,可以将参数转换为tensor数据类型,输入可以是序列【列表】、元组、ndarray,也可以是单独一个数等多种数据类型
torch.Tensor是一个类 输入参数需要是一个序列比如列表
https://blog.csdn.net/qq_36930266/article/details/104602792