たれぱんのびぼーろく

わたしの備忘録、生物学とプログラミングが多いかも

PyTorchの `Tensor.to`

キャストとデータ転送を担う関数。

device で転送先デバイスdtype で変換型を指定。既存Tensorを渡してそのdeviceとdtypeを参照させることも可能。
オプションは3つ:

  • non_blocking: 非同期のデータ転送
  • copy: キャスト/転送が実現済みが否かに関わらず新規Tensorを生成
  • memory_format: ?