dataset[batch_idx:, :, :, :]
这将收集3D数据批次。
dataset[batch_idx:, :]
这将收集一批一维数据。
我们如何收集n维数据批次？
寻找适用于numpy和h5py的解决方案
答案 0 :(得分：0)
我不确定h5py，但是对于numpy，我始终使用degs % 360。
slice每个slices = tuple([batch_idx] + [slice(None)]*(dim))
dataset[slices]
相当于一个分号，因此您只需要列出一个适合您所拥有的维数的列表即可。因此，如果dim为1，则上面的代码将收集一维数据的切片，如果dim为3，则将收集3维数据的切片，依此类推。如果要进行更复杂的切片，则可以使用{{ 1}}替换n维切片中的slice(None)。希望有帮助！