首页 > 编程知识 正文

squeeze python,pytorch 入门

时间:2023-05-04 04:48:17 阅读:111750 作者:169

#【PyTorch】squeeze函数详情

函数原型torch.squeeze(input,dim=None,*,out=None ) Tensor函数详细信息删除一个ssdsy中所有维数为1的维度

例如,(A 1 B C 1 D )维度) a )时间1 )时间)时间)时间) A 1 B C 1 D )的ssdsy调用ssy

如果指定dim参数,则squeeze操作仅应用于指定的维。 输入input的形状为(A 1 B ) a ) times1) A times B ) A 1 B )时,squeeze ) input,0 )不变更该ssdsy,但squeeze ) input,

示例x=torch.zeros (2,1,2,1,2 ) x.size ) Torch.size ) [ 2,1,2,1,2 ] ) y=torch.squeze ) x ) y.size

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。