摘要：
1 导入包 import torch import torch.nn as nn from torch.utils.data import TensorDataset,DataLoader from torch.nn import init import torch.optim as optim f 阅读全文
posted @ 2022-03-07 20:30
别关注我了，私信我吧
阅读(538)
评论(0)
推荐(1)
摘要：
1 导入实验需要的包 import torch from torch import nn import numpy as np import matplotlib.pyplot as plt from torch.utils.data import DataLoader,TensorDataset 阅读全文
posted @ 2022-03-07 19:25
别关注我了，私信我吧
阅读(839)
评论(0)
推荐(0)
摘要：
nn.Linear() PyTorch的 nn.Linear() 是用于设置网络中的全连接层的，需要注意在二维图像处理的任务中，全连接层的输入与输出一般都设置为二维张量，形状通常为[batch_size, size]，不同于卷积层要求输入输出是四维张量。其用法与形参说明如下： torch.nn.Li 阅读全文
posted @ 2022-03-07 19:10
别关注我了，私信我吧
阅读(5036)
评论(0)
推荐(0)
摘要：
1、二维矩阵乘法 torch.mm() torch.mm(mat1, mat2, out=None) 其中 $\operatorname{mat} 1 \in \mathbb{R}^{n \times m}, \operatorname{mat} 2 \in \mathbb{R}^{m \times 阅读全文
posted @ 2022-03-07 15:40
别关注我了，私信我吧
阅读(311)
评论(0)
推荐(0)
浙公网安备 33010602011771号