提交 df826de7 编写于 作者: L li099 提交者: whs

revise tensor array to tensor op (#14368)

test=develop
上级 6c329455
...@@ -235,11 +235,11 @@ def tensor_array_to_tensor(input, axis=1, name=None): ...@@ -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) 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 = helper.create_variable_for_type_inference(dtype=helper.input_dtype())
out_index = helper.create_variable_for_type_inference(dtype="int32") out_index = helper.create_variable_for_type_inference(dtype="int32")
helper.append_op( helper.append_op(
type='tensor_array_concat', type='tensor_array_to_tensor',
inputs={'X': input}, inputs={'X': input},
outputs={'Out': [out], outputs={'Out': [out],
'OutIndex': [out_index]}, 'OutIndex': [out_index]},
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册