torch.squeeze(input, dim=None, out=None) squeeze()函数的功能是维度压缩。返回一个tensor(张量),其中input 中大小为1的所有维都已删除。举个例子:如果input 的 ...
確定! 回上一頁