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成神之路。
共有 0 条评论