fluid.layer.shape 返回的数据类型问题
Created by: jiangjiajun
在文档中http://www.paddlepaddle.org/documentation/docs/zh/1.2/api_cn/layers_cn.html#permalink-149-shape 提到shape返回的是是int32类型的形状,但测试发现跟输入的数据类型一致(测试了int32的输入和float32的输入)。
查阅代码中看到输出的定义确实是与输入的数据类型一样 def shape(input): """ ${comment}
Args:
input (Variable): ${input_comment}
Returns:
out (Variable): ${out_comment}
"""
helper = LayerHelper('shape', **locals())
out = helper.create_variable_for_type_inference(
dtype=helper.input_dtype('input'))
helper.append_op(
type='shape', inputs={'Input': input}, outputs={'Out': out})
return out