torch中三维四维矩阵运算，以及多维softmax运算

最新推荐文章于 2025-04-25 20:33:30 发布

FIXLS

最新推荐文章于 2025-04-25 20:33:30 发布

阅读量7.4k

点赞数 1

分类专栏： Pytorch

本文链接：https://blog.csdn.net/baidu_15113429/article/details/104876995

版权

本文介绍了在PyTorch中如何进行三维矩阵的乘法运算，实验结果显示3*3*2的矩阵乘以2*3的矩阵得到3x3x3的结果。此外，还探讨了三维矩阵上的多维softmax计算，当dim=0时，是对0维度进行softmax操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

三维矩阵*三维矩阵


import torch

tensors = torch.tensor([[[1,2],[1,2],[1,2]],[[1,2],[1,2],[1,2]],[[1,2],[1,2],[1,2]]])
print(tensors.data)
print(tensors.data.shape)
mul_result = torch.matmul(tensors,tensors.transpose(1, 2))
print(tensors.transpose(1, 2))
print(tensors.transpose(1, 2).data.shape)
print(mul_result.data)
print(mul_result.data.shape)

实验结果：

tensor([[[1, 2],
         [1, 2],
         [1, 2]],

        [[1, 2],
         [1, 2],
         [1, 2]],

        [[1, 2],
         [1, 2],
         [1, 2]]])
torch.Size([3, 3, 2])
tensor([[[1, 1, 1],
         [2, 2, 2]],

        [[1, 1, 1],
         [2, 2, 2]],

        [[1, 1, 1],
         [2, 2, 2]]])
torch.Size([3, 2, 3])
tensor([[[5, 5, 5],
         [5, 5, 5],
         [5, 5, 5]],

        [[5, 5, 5],
         [5, 5, 5],
         [5, 5, 5]],

        [[5, 5, 5],
         [5, 5, 5],
         [5, 5, 5]]])
torch.Size([3, 3, 3])
<

最低0.47元/天解锁文章