pylops_gpu.utils.torch2numpy.torchtype_from_numpytype

pylops_gpu.utils.torch2numpy.torchtype_from_numpytype(numpytype)[source]

Convert torch type into equivalent numpy type

Parameters:
numpytype : numpy.dtype

Numpy type

Returns:
torchtype : torch.dtype

Torch equivalent type

Notes

Given limitations of torch to handle complex numbers, complex numpy types are casted into equivalent real types and the equivalent torch type is returned.