PyTorch
記事
- PyTorchのTensorの次元数、形状、要素数を取得: dim(), size(), numel()
- PyTorchのTensorの要素の値を取得: item()
- PyTorch Hub, torchvision.modelsで学習済みモデルをダウンロード・使用
- PyTorchの学習済みモデルで画像分類（VGG, ResNetなど）
- PyTorchでTensorとモデルのGPU / CPUを指定・切り替え
- PyTorchでGPU情報を確認（使用可能か、デバイス数など）
- PyTorchのTensorのデータ型（dtype）と型変換（キャスト）
- PyTorchでモデル（ネットワーク）を構築・生成
- PyTorchのtorch.flattenとtorch.nn.Flattenの違い