输出为元组类型
时间: 2024-02-24 18:58:41 浏览: 93
数组输出1
如果要将一个张量(tensor)转换为元组(tuple),可以使用 PyTorch 中的 `tuple()` 函数。示例代码如下:
```python
import torch
tensor = torch.tensor([1, 2, 3, 4, 5])
tup = tuple(tensor.tolist())
print(tup)
```
输出结果为:
```
(1, 2, 3, 4, 5)
```
注意,这里需要使用 `tolist()` 函数将张量转换为列表,再使用 `tuple()` 函数将列表转换为元组。
阅读全文