ensure_torch#

omnipose.core.ensure_torch(*arrays, device=None, dtype=torch.float32)[source]#

Convert numpy arrays to torch tensors if needed.