torch.nn模块介绍

torch.nn
CLASS torch.nn.Linear(in_features, out_features, bias=True) 输入： (N, *, in_features)，*代表可以拓展到任意大的维度 输出： (N, *, out_features), *含义是输出的除了out_features这个唯独外，所有的维度的形状都要和输入相同
学习的变量： 源码： 其中F指代functional模块
CLASS torch.nn.Softmax(dim=None)
对输入的n维数据进行softmax操作，对其进行尺度变换，使得n维数据的大小均位于[0,1]区间内并且加和为1.
其中，Softmax操作函数如下：

Softmax

(

torch.nn模块介绍最先出现在Python成神之路。