技术标签: Pytorch. squeeze的作用就是对tensor变量进行维度压缩,去除维数为1的的维度。 ☆例如,一个3×2×1×2×1的tensor,squeeze()之后便成了3×2×2。
確定! 回上一頁