tensor矩阵转置
时间: 2023-12-01 16:04:00 浏览: 113
矩阵的转置
tensor矩阵转置可以使用`t()`函数来实现。该函数可以将tensor矩阵的行数据转换为列数据,也可以将列数据转换为行数据。下面是一个例子:
```python
import torch
# 创建一个2x3的tensor矩阵
a = torch.tensor([[1, 2, 3], [4, 5, 6]])
# 对a进行转置操作
b = a.t()
# 输出转置后的结果
print(b)
```
输出结果为:
```
tensor([[1, 4],
[2, 5],
[3, 6]])
```
在上面的例子中,我们首先创建了一个2x3的tensor矩阵a,然后使用`t()`函数对其进行转置操作,最后输出转置后的结果。可以看到,转置后的结果是一个3x2的tensor矩阵,其中原来的第一行变成了第一列,原来的第二行变成了第二列,原来的第三行变成了第三列。
阅读全文