2022年5月14日
torch.squeeze
摘要： torch.squeeze(A，N) 如果不指定位置参数N，如果数组A的维度为（1，1，3）那么执行 torch.squeeze(A，1) 后A的维度变为 （1，3），中间的维度被删除 注： 1. 如果指定的维度大于1，那么将操作无效 2. 如果不指定维度N，那么将删除所有维度为1的维度 torch 阅读全文
posted @ 2022-05-14 13:07 今天记笔记了吗 阅读(490) 评论(0) 推荐(0)