共计 200 个字符，预计需要花费 1 分钟才能阅读完成。
Softmax
代码
import torch
import torch.nn as nn
softmax = nn.Softmax(dim=1)
input = torch.randn(1, 10)
output = softmax(input)
print(output.size())
torch.Size([1, 10])
援用
<1>
torch=1.7.1+cu101
torchvision=0.8.2
torchaudio=0.7.2
正文完
