Fix the init dtype.

Xreki · Xreki · commit d24c66c76090 · 2025-11-05T09:55:11.000+08:00
diff --git a/graph_net/paddle/utils.py b/graph_net/paddle/utils.py
@@ -213,7 +213,7 @@ def replay_tensor(info):
         ).to(device)
     else:
         if mean is not None and std is not None:
-            tensor = paddle.empty(shape=shape, dtype=dtype)
+            tensor = paddle.empty(shape=shape, dtype="float32")
             initializer = paddle.nn.initializer.TruncatedNormal(
                 mean=mean, std=std, a=min_val, b=max_val
             )

Original file line number	Diff line number	Diff line change
`@@ -213,7 +213,7 @@ def replay_tensor(info):`
`213`	`213`	`).to(device)`
`214`	`214`	`else:`
`215`	`215`	`if mean is not None and std is not None:`
`216`		`- tensor = paddle.empty(shape=shape, dtype=dtype)`
	`216`	`+ tensor = paddle.empty(shape=shape, dtype="float32")`
`217`	`217`	`initializer = paddle.nn.initializer.TruncatedNormal(`
`218`	`218`	`mean=mean, std=std, a=min_val, b=max_val`
`219`	`219`	`)`