torch. nn.Softmax(dim=2), 则对dim=2进行softmax,dim =0,1,2到底代表什么意思?
torch. nn.Softmax(dim=1)
最新推荐文章于 2024-07-29 20:28:46 发布
torch. nn.Softmax(dim=2), 则对dim=2进行softmax,dim =0,1,2到底代表什么意思?