MindSpore报错“ValueError: For ‘Conv2D‘, the x shape的size应是xx，但得到xx

当仁先生

于 2022-04-01 11:54:45 发布

阅读量1.5k

点赞数 2

文章标签：机器学习神经网络 python

本文链接：https://blog.youkuaiyun.com/weixin_45875914/article/details/123894327

版权

本文档记录了在MindSpore 1.6.0版本中遇到的一个错误，即在使用Conv2D算子时收到'ValueError: For ‘Conv2D’, the x shape的size应是xx，但得到xx'的错误信息。错误出现在GPU环境下，由于输入张量的shape不是预期的4维。解决方案是检查并修正输入张量的维度，确保与网络架构中其他参数匹配。经过调整，脚本成功运行。" 132437151,19776666,Vite Vue3 配置详解：vite.config.js,"['前端开发', 'Vue.js', 'Vite', '配置']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 报错描述

1.1 系统环境

Hardware Environment(Ascend/GPU/CPU): GPU

Software Environment:

– MindSpore version (source or binary): 1.6.0

– Python version (e.g., Python 3.7.5): 3.7.6

– OS platform and distribution (e.g., Linux Ubuntu 16.04): Ubuntu 4.15.0-74-generic

– GCC/Compiler version (if compiled from source):

1.2 基本信息

1.2.1 脚本

训练脚本是通过构建Conv2d的单算子网络，对输入张量计算二维卷积。脚本如下：

 01 class Net(nn.Cell):
 02   def __init__(self):
 03     super(Net, self).__init__()
 04     self.kernel = np.random.randn(

最低0.47元/天解锁文章