每天讲解一点PyTorch F.softmax
Posted cv.exp
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了每天讲解一点PyTorch F.softmax相关的知识，希望对你有一定的参考价值。
每天讲解一点PyTorch——3
现在我们学习F.softmax(x, dim = -1)，其中import torch.nn.functional as F
dim = -1表明对最后一维求softmax
>>> m
tensor([[0.1000, 0.2000],
[0.3000, 0.4000]])
>>> import torch.nn.functional as F
>>> y = F.softmax(m,dim=0) # 按列F.softmax,列和为1
>>> y
tensor([[0.4502, 0.4502],
[0.5498, 0.5498]])
>>>
>>> y = F.softmax(m,dim=1) #按行F.softmax,行和为1
>>> y
tensor([[0.4750, 0.5250],
[0.4750, 0.5250]])
>>>
以上是关于每天讲解一点PyTorch F.softmax的主要内容，如果未能解决你的问题，请参考以下文章