.unsqueeze(0)

仅作为记录,大佬请跳过

.unsqueeze(0)增加维度(0表示,在第一个位置增加维度)

.unsqueeze(0)_第1张图片
传送门

你可能感兴趣的:(python)