Python

【Pytorch】デバイス(CPU/CUDA)を指定する方法【GPU】 Pytorchでテンソルを扱うデバイスを指定するには、device引数や、.to()メソッドを用いる。 テンソルは以下に定義するものを用いる。 ...

【Pytorch】テンソルを連結する方法（cat・stack） Pytorchのテンソルを連結する方法をまとめておく。 連結方法としてはcatとstackがある。 既存のdimに沿ってテンソルを連結する（cat）...

【Pytorch】テンソルを分割する方法(split・chunk) Pytorchで宣言したテンソルの分割を行う方法をまとめる。 分割する前のテンソルを以下のコードで定義しておく。 a = torch.arang...

【Pytorch】テンソルの次元を入れ替え・変形する方法(reshape・transpose・permute) Pytorchで定義したテンソルの次元を入れ替えたり変形する方法をまとめておく。 入れ替え・変形にはres...

【Pytorch】テンソルの次元を追加・削除する方法【dim】 Pytorchでテンソル(Tensor)の次元を追加したり削除したりする方法をコード例でまとめておく。 元のテンソルとして以下を用いる。 0～５を要...

【Pytorch】テンソルのサイズを確認する方法(size・shape) PytorchではTensor（テンソル）のサイズを確認するための方法として２種類ある。 .size()と.shapeで調べることが出来る。 ...

【Pytorch】テンソルの型を定義する方法（dtype） Pytorchで定義したテンソルの型は、dtypeの引数を指定することで定義することができる。 テンソルの要素をfloat型にしたいときは以下のようにコードを...

【Pytorch】確率分布(標準正規・一様・ベルヌーイ・多項)の実装コード 乱数を初期化する（seed） 以下のコードで引数を任意の数値に固定すると、毎回同じ値が生成される。 torch.manual_seed(...

【Pytorch】基本的な行列を作る（宣言）する方法【Python】 Pytorchを使って行列を作る（宣言）する方法をまとめておく。 ここではサンプルプログラムを紹介する。 基本的な行列を作成するコード ...

【Adam】パラメータ最適化アルゴリズムAdamを実装するコード【Python】 Adamクラスの定義 Adamでの重み更新は以下のような式になる。 これを実装すると以下のようなコードになる。 ...