提交 d1a6ee20 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Cosmetic changes

上级 31787931
......@@ -2840,17 +2840,16 @@ class Join(Op):
as_tensor_variable_args = [unbroadcast(x, *range(x.type.ndim)) for x in as_tensor_variable_args]
inputs = [as_tensor_variable(axis)] + as_tensor_variable_args
if inputs[0].type not in int_types:
if inputs[0].type not in int_types:
raise TypeError('Axis could not be cast to an integer type', axis, inputs[0].type, int_types)
outputs = [output_maker(bcastable)]
node = Apply(self, inputs, outputs)
if any(not x.type.broadcastable[0] for x in orig):
node.tag.shape_zero = None
node.tag.shape_zero = None
else:
node.tag.shape_zero = len(orig)
#backport node.tag.shape_zero = None if any(not x.type.broadcastable[0] for x in orig) else len(orig)
node.tag.shape_zero = len(orig)
return node
def perform(self, node, axis_and_tensors, (out, )):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论