revert a+b axis change; test=develop (#21821)

1.6.2
hong 5 years ago committed by GitHub
parent 6624ea7673
commit 957e182a90
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -225,13 +225,15 @@ def monkey_patch_variable():
other_var = tmp
out = create_new_tmp_var(current_block(self), dtype=lhs_dtype)
axis = -1
if other_var.shape[0] == -1:
axis = 0
current_block(self).append_op(
type=op_type,
inputs={'X': [self],
'Y': [other_var]},
outputs={'Out': out},
attrs={'axis': -1})
attrs={'axis': axis})
return out
comment = OpProtoHolder.instance().get_op_proto(op_type).comment

Loading…
Cancel
Save