aiqimen4016的博客

转载 Pytorch 中的 dim

Pytorch 中对 tensor 的很多操作如 sum、argmax、unsqueeze 等都可以设置 dim 参数用来指定操作在哪一维进行。Pytorch 中的 dim 类似于 numpy 中的 axis，这篇文章来总结一下 Pytorch 中的 dim 操作。dim 与方括号的关系创建一个矩阵a = torch.tensor([[1, 2], [3, 4]])print(a...