PyTorch

関連記事

CPU 版と GPU 版の違い

CPU 版と GPU 版で以下の実行結果が異なるのがわかる。これは CPU 版と GPU 版ではより低レイヤーでソートアルゴリズムが異なることに起因していると自分は推測しているが不明である。 
import torch
x = torch.tensor([0.3, 0.2, 0.1, 0.4, 0.5])
print(x.topk(3, sorted=False))

# GPU
torch.return_types.topk(
values=tensor([0.5000, 0.4000, 0.3000]),
indices=tensor([4, 3, 0]))

# CPU
torch.return_types.topk(
values=tensor([0.4000, 0.5000, 0.3000]),
indices=tensor([3, 4, 0]))
Python