Q&AX(変数)に一番近い値のインデックスを取得する関数はありませんか？（pytorchのtensor型です)
探そうとしたんですがなんて検索したらいいかわかりません
もしあったら教えてくださいなければ
一番小さい値のインデックスを取得する関数でもいいです
回答2件
0
まさにこちらですね。
Qiitaに載ってました。
Python
1import numpy as np 2lst = [4, 7, 9] 3num = 5 4idx = np.abs(np.asarray(lst) - num).argmin() 5print(idx)
投稿2020/07/27 20:32
編集2020/07/27 20:38
総合スコア2720
0
自己解決
idx = np.abs(np.asarray(lst) - num).argmin()
に出てくる関数名を使って検索したら 何とかpytorchでの解決策がでました
投稿2020/07/28 20:59
総合スコア123
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/07/28 20:18
2020/07/28 22:14