2021-03-08から1日間の記事一覧

備忘録～ .data[0] でエラーが出たとき

PyTorch 0.5 以降、ゼロ次元テンソルから値を取り出すには .data[0] ではなく .item() を使う。