unienv_interface.backends.pytorch

PyTorchComputeBackend module-attribute

PyTorchComputeBackend: ComputeBackend = XBPytorchBackend

PyTorchArrayType module-attribute

PyTorchArrayType = torch.Tensor

PyTorchDeviceType module-attribute

PyTorchDeviceType = Union[torch.device, str]

PyTorchDtypeType module-attribute

PyTorchDtypeType = torch.dtype

PyTorchRNGType module-attribute

PyTorchRNGType = torch.Generator