言語
PyTorch クイックリファレンス
PyTorch クイックリファレンス の印刷用チートシート。すばやい参照、学習、机上利用向けに整理されています
サンプル
このチートシートの内容
- テンソル
- 自動微分
- ニューラルネットワーク
- データのロード
- 学習ループ
- オプティマイザ
- 損失関数
- 保存とロード
- GPU
- よくあるパターン
テンソルの作成
import torch
a = torch.tensor([1, 2, 3])
b = torch.zeros(2, 3)
c = torch.ones(3, 3)
d = torch.randn(2, 4) # normal dist
テンソルコンストラクタ
torch.zeros(m, n) | 全ゼロ、形状(m, n) |
torch.ones(m, n) | 全1、形状(m, n) |
torch.randn(m, n) | 標準正規分布のランダム値 |
torch.arange(start, end, step) | 等間隔の値 |
torch.linspace(start, end, steps) | 固定数の点 |
torch.eye(n) | 単位行列 |
torch.empty(m, n) | 未初期化メモリ |
NumPyとの相互変換
t = torch.from_numpy(np_array)
arr = tensor.numpy() # shares memory
t = torch.as_tensor(np_array)
すべてのセクションを含む印刷用チートシートをダウンロード
PDF をダウンロードダウンロード