torch.nn.LogSoftmax()与输入具有相同尺寸和形状的张量,其值在[-inf,0)范围内

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)

Next  Previous


版权声明:本文为studyvcmfc原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。