fix expand op dtype build bugs; test=develop (#18932)

f745d6d9 · hong · GitHub · 02c370c3 · f745d6d9
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +2 -1

未找到文件。
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -9690,7 +9690,8 @@ def expand(x, expand_times, name=None):
                    new_expand_times.append(ele)
                else:
                    assert (isinstance(ele, int))
-                    temp_out = helper.create_variable_for_type_inference(dtype)
+                    temp_out = helper.create_variable_for_type_inference(
+                        "int32")
                    fill_constant(
                        [1], 'int32', ele, force_cpu=True, out=temp_out)
                    new_expand_times.append(temp_out)