2025年2月15日
pytorch函数
摘要： squeeze() 和 unsqueeze()函数 torch.squeeze(A，N) torch.unsqueeze()函数的作用减少数组A指定位置N的维度，如果不指定位置参数N，如果数组A的维度为（1，1，3）那么执行 torch.squeeze(A，1) 后A的维度变为 （1，3），中间的维 阅读全文
posted @ 2025-02-15 20:55 ziv80 阅读(37) 评论(0) 推荐(0)