revise tensor array to tensor op (#14368)

test=develop
panyx0718-patch-1
li099 7 years ago committed by whs
parent 6c32945556
commit df826de76e

@ -235,11 +235,11 @@ def tensor_array_to_tensor(input, axis=1, name=None):
output, output_index = fluid.layers.tensor_array_to_tensor(input=tensor_array)
"""
helper = LayerHelper('tensor_array_concat', **locals())
helper = LayerHelper('tensor_array_to_tensor', **locals())
out = helper.create_variable_for_type_inference(dtype=helper.input_dtype())
out_index = helper.create_variable_for_type_inference(dtype="int32")
helper.append_op(
type='tensor_array_concat',
type='tensor_array_to_tensor',
inputs={'X': input},
outputs={'Out': [out],
'OutIndex': [out_index]},

Loading…
Cancel
Save