Update botorch/optim/utils/numpy_utils.py

pytorch · Nov 6, 2024 · 3557da3 · 3557da3
1 parent f6f11f9
commit 3557da3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/botorch/optim/utils/numpy_utils.py b/botorch/optim/utils/numpy_utils.py
@@ -124,7 +124,9 @@ def set_tensors_from_ndarray_1d(
             try:
                 size = tnsr.numel()
                 vals = array[index : index + size] if tnsr.ndim else array[index]
-                tnsr.copy_(torch.from_numpy(vals.reshape(tnsr.shape)).to(tnsr))
+                torch.as_tensor(vals, device=tnsr.device, dtype=tnsr.dtype).view(
+                    tnsr.shape
+                )
                 index += size
             except Exception as e:
                 raise RuntimeError(