pytorch 中的 @ 和 * 运算符

最新推荐文章于 2025-04-24 13:41:20 发布

原创最新推荐文章于 2025-04-24 13:41:20 发布 · 3.2k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #人工智能

在PyTorch中，`@`运算符表示矩阵乘法，遵循[m,n]和[n,k]的维度规则，产生[m,k]的矩阵。而`*`运算符用于执行Hadamard积，即两个相同形状的矩阵对应元素相乘。`t()`函数则用于转置二维矩阵。

部署运行你感兴趣的模型镜像

pytorch中的@表示的是数学中的矩阵乘法，*是数学中的Hadamard积（哈达玛积）

import torch
a = torch.tensor([[1,2],
                  [2,3],
                  [5,6]])
b = torch.tensor([[2,1],
                  [8,5],
                  [3,2]])
c = a*b
d = a @ b.t() ## [3,2] @ [2,3]
print('*',c)
print('@',d)

输出
在这里插入图片描述

做矩阵乘法时，要求两个矩阵的形状为[m,n]和[n,k],得到的是一个[m,k]的矩阵。
做Hadamarda积是，要求两个矩阵是同型矩阵，a和b都是2*3的。相同位置元素进行相乘

在pytorch中.t()表示二维矩阵的转置
参考

您可能感兴趣的与本文相关的镜像

PyTorch 2.9

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理