目录
逐元素相乘和点乘 2024.05.20笔记 需要确认
两个向量点乘是对应位置相乘再求和,结果是一个数字。
逐元素相乘:是对应位置相乘,结果是相同维度向量
矩阵乘号是点乘,自动扩展
import torch
# 创建张量 a 和 b,使用确定的整数序列
a = torch.tensor([[[[[
1, 1],
[1, 1]
]]]]).expand(1, 1, 2, 2, 2)
b = torch.tensor([[[[[
1, 2],
[3, 4]
]]]]).expand(1, 1, 1, 2, 2)
# 逐元素相乘
result = a * b
# 输出结果
print("a:\n", a)
print("b:\n", b)
print("result:\n", result)
这篇博客主要探讨了PyTorch中的点乘操作,包括两个向量如何进行点乘,即对应位置相乘再求和得到一个标量结果,以及逐元素相乘的区别。此外,还提到了在PyTorch中,即使向量维度不同,只要子维度匹配,仍可进行点乘。不过,博主指出,如果不注意维度匹配,可能会导致错误。
订阅专栏 解锁全文
6544





