PyTorch中dim的含义
在对张量操作的过程中,dim的含义尤为重要,参考了两篇博客的讲解,现梳理如下视作对某个维度操作假设定义了某张量aa = torch.tensor([[1,4],[3,2]])很容易看出它的形状是2×2的,即有两个维度:第零个维度包含[1,4], [3,2]两个tensor,第一个维度包含了4个(两对)标量。若想沿某个维度找到最大值,则执行torch.max(a, dim)若令dim=0,则是从第零个维度中找到最大值。tensor([[1, 4], [3, 2]])张量
原创
2021-10-20 18:15:51 ·
472 阅读 ·
0 评论