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