pytorch中张量相乘

在PyTorch中，可以使用torch.matmul()函数进行张量相乘。如果要进行逐元素相乘，则可以使用"*"运算符。

示例代码：

import torch

# 定义两个张量
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])

# 张量相乘
c = torch.matmul(a, b)
print(c)

# 逐元素相乘
d = a * b
print(d)

输出结果：

tensor([[19, 22],
        [43, 50]])
tensor([[ 5, 12],
        [21, 32]])