提交 d92a95d2 authored 作者: AndroidCloud's avatar AndroidCloud

updates

上级 055d6152
...@@ -4868,11 +4868,9 @@ class Reshape(Op): ...@@ -4868,11 +4868,9 @@ class Reshape(Op):
if isinstance(node.inputs[0], TensorVariable): if isinstance(node.inputs[0], TensorVariable):
x, shp = inputs x, shp = inputs
z, = outputs z, = outputs
new_ndim = self.ndim
sdtype = node.inputs[1].type.dtype_specs()[1] sdtype = node.inputs[1].type.dtype_specs()[1]
fail = sub['fail'] fail = sub['fail']
params = sub['params'] params = sub['params']
return """ return """
assert (PyArray_NDIM(%(shp)s) == 1); assert (PyArray_NDIM(%(shp)s) == 1);
npy_intp new_dims[%(params)s->ndim]; npy_intp new_dims[%(params)s->ndim];
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论