pytorch源码:https://pytorch.org/docs/stable/index.html
二维卷积:
激活函数:
nn.ReLU(): max(features, 0)
池化层:
class torch.nn.MaxPool2d(kernel_size, stride=None, padding=0, dilation=1,
return_indices=False, ceil_mode=False)
容器:
nn.Sequential()
全连接层:
归一化:
torch.max(input, dim) 函数
output = torch.max(input, dim)
输入
input是softmax函数输出的一个tensor
dim是max函数索引的维度0/1,0是每列的最大值,1是每行的最大值
输出
函数会返回两个tensor,第一个tensor是每行的最大值;第二个tensor是每行最大值的索引。