python:怎么把两行三列的torch.Tensor转置三行两列的numpy.ndarray
时间: 2024-02-11 15:07:43 浏览: 78
浅谈pytorch中torch.max和F.softmax函数的维度解释
可以使用 `t()` 函数将 `torch.Tensor` 转置,然后再使用 `numpy()` 函数将其转换为 `numpy.ndarray`,示例代码如下:
```python
import torch
import numpy as np
# 创建一个 2x3 的 tensor
tensor = torch.tensor([[1, 2, 3], [4, 5, 6]])
# 将 tensor 转置并转换为 3x2 的 numpy 数组
array = tensor.t().numpy()
print(array)
```
输出结果为:
```
array([[1, 4],
[2, 5],
[3, 6]])
```
需要注意的是,`t()` 函数返回的是一个新的张量,不会改变原来的张量。如果要在原来的张量上进行转置操作,可以使用 `transpose()` 函数。
阅读全文