您的位置:首页 > 产品设计 > UI/UE

AssertionError: HybridBlock requires the first argument to forward be either Symbol or NDArray, but

2018-03-16 16:02 961 查看
mxnet中的HybridBlock只接受Symbol和NDArray数据,如果你送入的是numpy.ndarray或者list数据,就会报如上的错误。

解决办法:把数据变为NDArray就行了。

>>> import mxnet as mx
>>> import numpy as np
>>> data = np.arange(5)
>>> b = mx.nd.array(data)      # 转换方法
>>> b
[ 0.  1.  2.  3.  4.]
<NDArray 5 @cpu(0)>
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐