如何将torch.utils.data.dataset.TensorDataset object打印
时间: 2023-12-12 18:03:12 浏览: 120
python torch.utils.data.DataLoader使用方法
5星 · 资源好评率100%
如果要将 `torch.utils.data.dataset.TensorDataset` 对象打印出来,可以使用 `print()` 函数打印该对象的属性。例如,可以打印该对象的 `tensors` 属性,该属性是一个元组,包含了数据集的输入数据和标签数据。下面是一个简单的示例代码:
```
import torch
from torch.utils.data import TensorDataset
# 创建一个 TensorDataset 对象
data = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
label = torch.tensor([0, 1, 0])
dataset = TensorDataset(data, label)
# 打印 TensorDataset 对象
print(dataset.tensors)
```
运行以上代码将输出如下结果:
```
(tensor([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]), tensor([0, 1, 0]))
```
可以看到,该代码输出了 `TensorDataset` 对象的 `tensors` 属性,包含了输入数据和标签数据。
阅读全文