unienv_interface.backends.pytorch

PyTorchComputeBackend module-attribute

PyTorchComputeBackend: ComputeBackend = PyTorchComputeBackend

PyTorchArrayType module-attribute

PyTorchArrayType = Tensor

PyTorchDeviceType module-attribute

PyTorchDeviceType = Union[device, str]

PyTorchDtypeType module-attribute

PyTorchDtypeType = dtype

PyTorchRNGType module-attribute

PyTorchRNGType = Generator