假设a是一个(2,3,4)形状的矩阵,a.unsqueeze(0)会在第一维度前插入一个维度,为(1,2,3,4),数据总量(维度相乘的结果)不会改变
unsqueeze方法
最新推荐文章于 2024-05-07 10:25:01 发布
假设a是一个(2,3,4)形状的矩阵,a.unsqueeze(0)会在第一维度前插入一个维度,为(1,2,3,4),数据总量(维度相乘的结果)不会改变