PyTorch torch.squeeze 関数
Pytorch torch リファレンスマニュアルtorch.squeezePyTorch において、サイズが 1 の次元をテンソルから削除するために使用される関数です。
関数定義
torch.squeeze(input, dim)
使用例
例
import torch
# サイズ1の次元を持つテンソルを作成
x = torch.randn(1, 2, 1, 3, 1)
print(元の形状：, x.shape)
# サイズ1の次元をすべて削除
y = torch.squeeze(x)
print(スクイーズ後の形状：, y.shape)
# 次元を指定
z = torch.squeeze(x, dim=0)
print(dim=0 スクイーズ後：, z.shape)
# サイズ1の次元を持つテンソルを作成
x = torch.randn(1, 2, 1, 3, 1)
print(元の形状：, x.shape)
# サイズ1の次元をすべて削除
y = torch.squeeze(x)
print(スクイーズ後の形状：, y.shape)
# 次元を指定
z = torch.squeeze(x, dim=0)
print(dim=0 スクイーズ後：, z.shape)
出力結果は次のとおりです：
原始形状: torch.Size([1, 2, 1, 3, 1]) 挤压后形状: torch.Size([2, 3]) dim=0 挤压后: torch.Size([2, 1, 3, 1])
その他の拡張