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) 求指定维度上的范数