LOGSOFTMAX
torch.nn.
LogSoftmax
(dim=None)[SOURCE]
Shape:
输入: (*) ( * ) ,其中*表示任意数量的附加尺寸
输出: (*) ( ∗ ) ,形状与输入相同
Parameters
dim ( int )–将用来计算LogSoftmax的维.
Returns
与输入具有相同尺寸和形状的张量,其值在[-inf,0)范围内
Examples:
>>> m = nn.LogSoftmax() >>> input = torch.randn(2, 3) >>> output = m(input)
版权声明:本文为studyvcmfc原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。