torch. nn.Softmax() dim的含义

最新推荐文章于 2024-09-07 14:53:28 发布

tsz danger

最新推荐文章于 2024-09-07 14:53:28 发布

阅读量4.6k

点赞数 2

分类专栏： GAN 文章标签：深度学习 pytorch

本文链接：https://blog.csdn.net/weixin_43914889/article/details/104505512

版权

GAN 专栏收录该内容

31 篇文章 8 订阅

订阅专栏

dim实际上是对不同维度进行求对数均值
例如下面这个是(3,2)的一个变量，dim = 0 实际上是对第一维的3个变量进行对数化，而dim = 1是对第二维进行操作

a = torch.Tensor([[1,1],[2,2],[3,3]])
a.size()
Out[89]: torch.Size([3, 2])
b = torch.nn.Softmax(dim=0)(a)
b
Out[91]: 
tensor([[0.0900, 0.0900],
        [0.2447, 0.2447],
        [0.6652, 0.6652]])
b = torch.nn.Softmax(dim=1)(a)
b
Out[93]: 
tensor([[0.5000, 0.5000],
        [0.5000, 0.5000],
        [0.5000, 0.5000]])