1、transform.ToTensor():转变成torch张量
2、view(输入通道数,输出通道数,宽带,高度):变成一个序列
3、torch.nn.Conv2d(in_channels,out_channels,kernel_size=kernel_size)
4、torch.nn.rnn(input_size,hidden_size,num_layer)
5、torch.mul(a,b)是对位相乘,a,b的维度一定要相等,乘出来维度不变
torch.mm(a,b)是矩阵乘法,就是那种一行乘以一列
torch.matmul()不规定大小和维度,具有广播机制对不同维度进行相乘操作
Pytorch入门记录
最新推荐文章于 2025-12-19 17:51:13 发布
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
4万+

被折叠的 条评论
为什么被折叠?



