2017-12-30 30 views
2

Pytorchには、IntTensorまたはcuda.IntTensorの変数があります。 64ビットのcpu/gpuを保持するように変更する必要があります。 variable.long()は、pytorchのすべての実装で64ビットを保証しますか?そうでない場合、どのようにして変数を64ビットに変換してすべての実装でcpu/gpuを保持できますか?pytorch:variable.long()は64ビットを保証しますか?

答えて

0

PyTorchのドキュメントから、torch.LongTensorまたはtorch.cuda.LongTensorを使用して、64ビット符号付き整数を保証することができます。

同様に、64ビット浮動小数点の場合は、torch.DoubleTensorまたはtorch.cuda.DoubleTensorを使用できます。

Variable(tensor)メソッドを使用して変数を変数に変換できます。ここで、tensorは、上に作成したテンソルを表します。

+0

質問を再読してください。 – BSalita

関連する問題