pytorch 张量乘法

PyTorch中可以使用`torch.matmul()`函数进行张量乘法。如果需要进行逐元素乘法，则可以使用`torch.mul()`函数。

下面是一个示例代码，展示如何使用`torch.matmul()`进行张量乘法：

import torch

# 创建两个张量
x = torch.randn(2, 3)
y = torch.randn(3, 4)

# 进行张量乘法
z = torch.matmul(x, y)

# 打印结果
print(z)

输出：

tensor([[ 0.5614,  0.2957,  0.8999, -0.0384],
        [-0.6807, -1.6377,  0.8762, -0.4621]])

如果需要进行逐元素乘法，可以使用`torch.mul()`函数，示例如下：

import torch

# 创建两个张量
x = torch.randn(2, 3)
y = torch.randn(2, 3)

# 进行逐元素乘法
z = torch.mul(x, y)

# 打印结果
print(z)

输出：

tensor([[-0.3223,  0.6475,  0.3215],
        [-0.0222,  0.1238,  0.0279]])