pytorch 中的 @ 和 * 运算符

huahuahuahhhh

于 2023-06-17 16:51:51 发布

阅读量3.1k

点赞数 2

CC 4.0 BY-SA版权

文章标签： pytorch python 人工智能

本文链接：https://blog.youkuaiyun.com/Blankit1/article/details/131262108

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

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

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()表示二维矩阵的转置
参考