torch.squeeze()和torch.unsqueeze()
摘要:
原文:https://www.jianshu.com/p/e0ddfcf2e72d 1. torch.squeeze(tensor) 和numpy等库函数中的squeeze()函数作用一样,torch.squeeze()函数的作用是压缩一个tensor的维数为1的维度,使该tensor降维变成最紧凑 阅读全文
posted @ 2019-11-30 21:15 KAKAFEIcoffee 阅读(1155) 评论(0) 推荐(0) 编辑