在Tensor上的索引和切片一般索引從前往後提供的索引,會依次在每個維度上做索引。 import torch a = torch.rand(4, 3, 28, 28) print(a[0].shape) ...
確定! 回上一頁