Data type and device transfers

metatrain.utils.transfer.batch_to(systems: List[System], targets: Dict[str, TensorMap], extra_data: Dict[str, TensorMap] | None = None, dtype: dtype | None = None, device: device | None = None)[source]

Changes the systems and targets to the specified floating point data type.

Parameters: