pytorch_pfn_extras.as_numpy_dtype

pytorch_pfn_extras.as_numpy_dtype(torch_dtype)

Returns NumPy dtype for the given PyTorch dtype.

Parameters:

torch_dtype (dtype) – PyTorch’s dtype object.

Returns:

NumPy type object.

Return type:

Any