たれぱんのびぼーろく

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

2023-07-07から1日間の記事一覧

PyTorchの `Tensor.to`

キャストとデータ転送を担う関数。 device で転送先デバイス、dtype で変換型を指定。既存Tensorを渡してそのdeviceとdtypeを参照させることも可能。 オプションは3つ: non_blocking: 非同期のデータ転送 copy: キャスト/転送が実現済みが否かに関わらず新規…