Q&A
tensor_aが[1,2,3,4,5]だとします
tensor_bが[1,0,0,4,5]だとします
tensor_aとtensor_bを比較して、一致しない要素のインデックスを取得するにはどうすれば良いですか？
この場合はインデックス1,2,3が一致しない場所だと分かればよいです。
list化してforで回すしかないのでしょうか？なるべくforを使わずにpytorchの機能などを使って解決したいです。
回答2件
0
putorchは詳しくありませんが、以下でできそうです。
python
1np.where(tensor_a.numpy() != tensor_b.numpy())
投稿2021/04/27 16:27
総合スコア24670
0
ベストアンサー
PyTorchの機能だけ使う解です。PyTorchを使い慣れると、「何でもtensorの世界で完結したくなりnumpyやskleanなどの機能をどうやって実現するかに頭を悩ませる病」に罹患してしまいます。私もです笑。
Python
1torch.where(tensor_a != tensor_b) 2# (tensor([1, 2]),)
この場合はインデックス1,2,3が一致しない場所だと分かればよいです。
1,2だけですよね。
投稿2021/04/28 23:34
編集2021/04/29 00:42
総合スコア3266
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。