2021-01-02から1日間の記事一覧
PyTorch でn次元のTensorがあったときに、n-1次元のTensorを使ってインデックスする方法です。 以下のように a というTensorと、 idx というインデックスがあったときに、 answer を取る方法です。 a = torch.tensor([ [100, 200, 300, 400], [500, 600, 70…
PyTorch でn次元のTensorがあったときに、n-1次元のTensorを使ってインデックスする方法です。 以下のように a というTensorと、 idx というインデックスがあったときに、 answer を取る方法です。 a = torch.tensor([ [100, 200, 300, 400], [500, 600, 70…