Pytorch中常见函数

Pytorch中常见函数

nn.Linear(input_feature, output_feature)
他的输入和输出需要满足的是：输入input(A, ..., input_feature)，输出output(B, ..., output_feature)

nn.Conv2D(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)

torch.norm(input, p = 'fro', dim=None, keepdim=False, out=None)
返回给定张量的矩阵范数或向量范数，
torch.norm(input,p=2),求二范数，输出是一个数；如果求一范数就是p=1
torch.norm(input, p, dim) 求指定维度上的范数