y-kamiya https://blog.hatena.ne.jp/y-kamiya/ MEMOcho- https://jsapachehtml.hatenablog.com/ python pytorch Double DQNの実装に必要になるちょっとした計算についてメモ ２つの２次元tensor x, yを用意し、"xの各行において最大の値を持つ要素"と同じ位置にあるyの要素を取得する >>> x = torch.rand(3,5) >>> x tensor([[ 0.0778, 0.6633, 0.4953, 0.1461, 0.4691], [ 0.3024, 0.0295, 0.3526, 0.6040, 0.7512], [ 0.1778, 0.7783, 0.1738, 0.5278, 0.0372]]) >>> x.argmax(1) tensor([ 1, 4, 1]) >>> … 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Fjsapachehtml.hatenablog.com%2Fentry%2F2018%2F08%2F26%2F142527" title="pytorchでindexのリストを指定してtensorの要素を取得 - MEMOcho-" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> Hatena Blog https://hatena.blog 2018-08-26 14:25:27 rich https://jsapachehtml.hatenablog.com/entry/2018/08/26/142527 1.0 100%