tensor = tensor[torch.randperm(tensor.size(0))] # Shuffle the first dimension ... torch.nonzero (tensor) # Index of non-zero elements torch.nonzero(tensor ...
確定! 回上一頁