🍔

torch.tensor.data はもう使わない

2022/06/19に公開

torch.tensor と torch.tensor.data の違いが分からなすぎて、ドキュメント巡回していたら1時間くらい溶かしてしまったのでメモしておきます。

torch.tensor.data とすると結局 tensor インスタンスが返ってきているっぽく

a = torch.rand(3, 5)
type(a)                      #  torch.Tensor
type(a.data)              #  torch.Tensor
id(a)                          #  4493764504
id(a.data)                 #  4493475488

のようになる。

https://discuss.pytorch.org/t/the-difference-between-torch-tensor-data-and-torch-tensor/25995
によると、.dataは後方互換性のために残されているだけの様子。
結論、torch.tensor を使うで良いっぽいです。

Discussion