fix bug of device

Author: Han Wang

source/tests/pt_expt/utils/test_network.py

@@ -165,7 +165,7 @@ def test_cross_backend_consistency(self) -> None:
         # Test forward pass
         rng = np.random.default_rng()
         x_np = rng.standard_normal((5, self.in_dim))
-        x_torch = torch.from_numpy(x_np)
+        x_torch = torch.from_numpy(x_np).to(env.DEVICE)
 
         out_dp = dp_net.call(x_np)
         out_pt = pt_net(x_torch).detach().cpu().numpy()