Enumerate Pytorch Tensor

1 2