Pytorch __getitem__()

__getitem__的作用是用来给予[]实现的,有两种方式:

  1. 通过[]去访问;
  2. 对对象进行访问,作为一个迭代对象访问,数组也是这么访问的,同时数组也是通过[]访问,所以实际可以把对象看做一个数组。

例如:dataloader本质上是一个可迭代对象,debug loader的时候就可以用__getitem__:

args = args()
loader = DataLoader(args)
images, labels = loader.__getitem__(1)
print(labels.shape)

DataLoader的作用就是将Dataset封装成一个Batch Size大小的Tensor,用于后面的训练。

你可能感兴趣的:(#,PyTorch,pytorch,人工智能,python)