pytorch : variable.long()이 64 비트를 보장합니까?

Question

pytorch에는 IntTensor 또는 cuda.IntTensor 일 수있는 변수가 있습니다. 64 비트를 유지하는 CPU/GPU로 변경해야합니다. variable.long()은 pytorch의 모든 구현에서 64 비트를 보장합니까? 그렇지 않다면, 변수는 모든 구현에서 CPU/GPU를 유지하면서 64 비트로 어떻게 변환 될 수 있습니까?

Answer 1

PyTorch 문서 에서 torch.LongTensor 또는 torch.cuda.LongTensor을 사용하면 64 비트 부호있는 정수를 사용할 수 있습니다.

마찬가지로 64 비트 부동 소수점의 경우 torch.DoubleTensor 또는 torch.cuda.DoubleTensor을 사용할 수 있습니다.

메서드를 사용하여 변수를 변수로 변환 할 수 있습니다. 여기서 tensor은 위에서 만든 텐서입니다.