PyTorch torch.unsqueeze 関数
Pytorch torch リファレンスマニュアルtorch.unsqueezePyTorch で指定位置にサイズ 1 の次元を挿入するための関数です。
関数定義
torch.unsqueeze(input, dim)
使用例
実例
import torch
x = torch.randn(2, 3)
print("元の形状:", x.shape)
# dim=0 に次元を挿入
y = torch.unsqueeze(x, dim=0)
print("dim=0 拡張後:", y.shape)
# dim=1 に次元を挿入
z = torch.unsqueeze(x, dim=1)
print("dim=1 拡張後:", z.shape)
x = torch.randn(2, 3)
print("元の形状:", x.shape)
# dim=0 に次元を挿入
y = torch.unsqueeze(x, dim=0)
print("dim=0 拡張後:", y.shape)
# dim=1 に次元を挿入
z = torch.unsqueeze(x, dim=1)
print("dim=1 拡張後:", z.shape)
出力結果は：
原始形状: torch.Size([2, 3]) dim=0 扩展后: torch.Size([1, 2, 3]) dim=1 扩展后: torch.Size([2, 1, 3])
その他の拡張