scatter_nd_add 返回 tensor 的 dtype 错误
Created by: AoZhang
-
环境 1）PaddlePaddle版本：1.6.2 2）CPU：Intel(R) Xeon(R) CPU E5-2620 v2 3）GPU：no 4）系统环境：CentOS 6.3 ，Python版本 3.6.5
-
复现信息：
x = fluid.layers.data(name='x', shape=[3, 3], dtype='float32', append_batch_size=False)
index = fluid.layers.data(name='index', shape=[2, 2], dtype='int64', append_batch_size=False)
update = fluid.layers.data(name='update', shape=[2], dtype='float32', append_batch_size=False)
output = fluid.layers.scatter_nd_add(x, index, update)
print(output.dtype)- 问题描述 print 输出的是 VarType.BOOL，正常不应该是 FP32 吗？但是我打印 output 的值发现是符合预期的。