pytorch 中的 @ 和 * 运算符

最新推荐文章于 2025-02-23 07:50:34 发布

DongHappyyy

最新推荐文章于 2025-02-23 07:50:34 发布

阅读量3.1w

点赞数 44

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

本文链接：https://blog.csdn.net/weixin_39715012/article/details/116380945

版权

torch 专栏收录该内容

6 篇文章

订阅专栏

本文详细介绍了在PyTorch中使用@和*操作符进行矩阵运算的方法。@操作符用于实现标准的矩阵乘法，而*操作符则用于两个矩阵对应元素之间的逐位乘法。通过具体实例展示了这两种运算的不同之处。

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

import torch
x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[2, 1], [4, 3]])
print("x_shape", x.shape)
print("y_shape", y.shape)

c = x@y
print("c_shape", c.shape)
print(c)


# 结果： 
x_shape torch.Size([2, 2])
y_shape torch.Size([2, 2])
c_shape torch.Size([2, 2])
tensor([[10,  7],
       [22, 15]])

import torch
x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[2, 1], [4, 3]])
print("x_shape", x.shape)
print("y_shape", y.shape)


d = x*y
print("d_shape", d.shape)
print(d)



# 结果：
x_shape torch.Size([2, 2])
y_shape torch.Size([2, 2])
d_shape torch.Size([2, 2])
tensor([[ 2,  2],
        [12, 12]])