本文共 261 字,大约阅读时间需要 1 分钟。
torch.nn.
LogSoftmax
(dim=None)
Shape:
输入: (*) ( * ) ,其中*表示任意数量的附加尺寸
输出: (*) ( ∗ ) ,形状与输入相同
Parameters
dim ( )–将用来计算LogSoftmax的维.
Returns
与输入具有相同尺寸和形状的张量,其值在[-inf,0)范围内
Examples:
>>> m = nn.LogSoftmax()>>> input = torch.randn(2, 3)>>> output = m(input)
转载地址:http://zvfti.baihongyu.com/