写一段Pytorch计算矩阵乘法的代码
时间: 2023-11-28 20:47:21 浏览: 84
矩阵乘法代码
可以使用 PyTorch 的 `matmul` 函数来实现矩阵乘法,示例代码如下:
``` python
import torch
# 定义两个矩阵
x = torch.randn(2, 3)
y = torch.randn(3, 4)
# 计算矩阵乘法
z = torch.matmul(x, y)
print(x)
print(y)
print(z)
```
其中,`torch.randn` 函数用于生成指定维度的随机数矩阵,`torch.matmul` 函数用于计算矩阵乘法。在这个示例中,先定义了两个矩阵 `x` 和 `y`,然后通过 `torch.matmul(x, y)` 计算它们的矩阵乘法结果,存储在变量 `z` 中,最后打印出来。
阅读全文