PyTorchでDataLoaderを使用して読み込んだデータの表示方法は？

PyTorchを使って、DataLoaderでデータセットを読み込んだ後、イテレータを使ってデータセットのデータを確認することができます。DataLoaderオブジェクトをイテレータに変換するにはiter()関数を使い、次のバッチのデータを取得するにはnext()関数を使います。

以下是一个示例代码，演示了如何使用DataLoader读取和查看数据。

import torch
from torch.utils.data import DataLoader

# 假设有一个名为dataset的数据集对象

# 创建一个DataLoader对象
dataloader = DataLoader(dataset, batch_size=32, shuffle=True)

# 将DataLoader对象转换为迭代器
data_iter = iter(dataloader)

# 获取下一个batch的数据
data = next(data_iter)

# 打印数据的shape
print(data.shape)

コードの上では、最初にDataLoaderオブジェクトが作成され、batch_sizeやシャッフルの有無が指定されます。次に、iter()関数を使用してDataLoaderオブジェクトをイテレーターに変換し、次にnext()関数を使用して次のバッチのデータを取得します。最後にデータの形状を出力してデータを確認します。

next()関数を使って得られたデータはテンソルオブジェクトであり、.numpy()メソッドを使うことでNumPy配列に変換することができ、データの内容をより簡単に確認することができます。

# 将Tensor对象转换为NumPy数组
data_numpy = data.numpy()

# 打印数据内容
print(data_numpy)

DataLoaderが読み込んだデータを確認できます。