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成神之路

版权声明:
作者:玉兰
链接:https://www.techfm.club/p/11970.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>