torch.unsqueeze(img1, dim=0)
时间: 2023-04-23 11:01:52 浏览: 195
浅谈pytorch torch.backends.cudnn设置作用
5星 · 资源好评率100%
torch.unsqueeze(img1, dim=0) 的作用是在 img1 的维度 dim 上增加一个维度,使得 img1 变成一个四维张量。具体来说,如果 img1 的形状为 (C, H, W),则 torch.unsqueeze(img1, dim=0) 的形状为 (1, C, H, W)。其中,1 表示新增的维度。
阅读全文