提交 0011ec8a authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Iterate over node.inputs, so they are already casted as Variables.

This removes the need for a special 'ndarray' case, and should generalize to Python and numpy numeric types, as well as CudaNdarray, and even other types.
上级 3e1516a9
...@@ -331,11 +331,9 @@ class PureOp(object): ...@@ -331,11 +331,9 @@ class PureOp(object):
# build test input-values # build test input-values
input_vals = [] input_vals = []
for ins in inputs: for ins in node.inputs:
if isinstance(ins, graph.Constant): if isinstance(ins, graph.Constant):
input_vals.append(ins.value) input_vals.append(ins.value)
elif isinstance(ins,numpy.ndarray):
input_vals.append(ins)
elif isinstance(ins,SharedVariable): elif isinstance(ins,SharedVariable):
input_vals.append(ins.get_value(borrow=True)) input_vals.append(ins.get_value(borrow=True))
elif isinstance(ins,graph.Variable) and hasattr(ins.tag, 'test_value'): elif isinstance(ins,graph.Variable) and hasattr(ins.tag, 'test_value'):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论