The former produces a tensor, which is recommended. for i, batch in enumerate(dataset): if i == 255 or i == 256: print(i, batch.shape) ...
確定! 回上一頁