目录
逐元素相乘和点乘 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)