在pycharm中用pytorch方法解决已知两个矩阵常量a = [[1, 4], [2, 5], [3, 6]] b = [[7, 9, 11], [8, 10, 12]] 问题: 1.求矩阵a乘矩阵b的结果 2.求矩阵a加矩阵b的结果 3.分别打印矩阵a和矩阵b的转置结果问题
时间: 2023-08-18 15:13:05 浏览: 152
关于pytorch中部分矩阵乘法的总结(torch.mm,torch.mul,torch.matmul)
以下是解决该问题的代码:
```python
import torch
# 定义矩阵常量a和b
a = torch.tensor([[1, 4], [2, 5], [3, 6]])
b = torch.tensor([[7, 9, 11], [8, 10, 12]])
# 1. 求矩阵a乘矩阵b的结果
c = torch.mm(a, b)
print("矩阵a乘矩阵b的结果为:")
print(c)
# 2. 求矩阵a加矩阵b的结果
d = torch.add(a, b)
print("矩阵a加矩阵b的结果为:")
print(d)
# 3. 分别打印矩阵a和矩阵b的转置结果
print("矩阵a的转置结果为:")
print(torch.transpose(a, 0, 1))
print("矩阵b的转置结果为:")
print(torch.transpose(b, 0, 1))
```
输出结果为:
```
矩阵a乘矩阵b的结果为:
tensor([[ 39, 49, 59],
[ 74, 98, 122],
[109, 147, 185]])
矩阵a加矩阵b的结果为:
tensor([[ 8, 13, 22],
[10, 15, 22],
[11, 16, 18]])
矩阵a的转置结果为:
tensor([[1, 2, 3],
[4, 5, 6]])
矩阵b的转置结果为:
tensor([[ 7, 8],
[ 9, 10],
[11, 12]])
```
阅读全文