2020-01-01から1年間の記事一覧

2つのTensor型配列を走査し一致している要素数を取得する

参考記事:stackoverflow.com 2つのTensor型配列AとBについて、まず "A == B" によって、要素が一致しているインデックスはTrue、一致していないインデックスはFalseとなっている配列を取得する。次に、"(A == B).sum()" または "torch.sum(A == B)" によっ…

list型とTensor型のシャッフル方法

目次 list型のシャッフル方法 random.shuffle()を使う random.sample()を使う PyTorchのTensor型のシャッフル list型のシャッフル方法 random.shuffle()を使う random.shuffle()の引数にリストをセットすると、リストの内容がシャッフルされます。 破壊的メ…

Google Colabでmakeするときのjオプションの最適な値

結論から言うと、ColabのCPUおよびGPUのコア数2を指定してmake -j2またはmake -j`nproc`とするのが一番速そうです。 ・・・ 最近Google Colabを使ってあるものを作っています。そこでは使用する前に長時間のビルドを要するツールをいくつか使っているのです…