每天讲解一点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的主要内容,如果未能解决你的问题,请参考以下文章

每天讲解一点PyTorch torch.matmul

每天讲解一点PyTorch torch.matmul

每天讲解一点PyTorch isinstance

每天讲解一点PyTorch isinstance

每天讲解一点PyTorch 12enumerate

每天讲解一点PyTorch 12enumerate