Make組ブログ

Python、Webアプリや製品・サービス開発についてhirokikyが書きます。

2021-01-02から1日間の記事一覧

PyTorchでn次元のTensorからn-1次元のTensorを使ってインデックスする

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