摘要： 1、nn.Conv2d class torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True) 二维卷积层, 输入的尺度是(N, C_in, 阅读全文

摘要： 1、squeeze()函数和unsqueeze()函数 首先要知道tensor的维度，比如tensor([[0, 1, 2],[ 3, 4, 5]])，维度是(2, 3)，当其维度变为(2, 1, 3)时，代表2个1行3列的矩阵，为tensor([[[0, 1, 2],[[3, 4, 5]]])。 阅读全文