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 (torch.dtype) – PyTorch’s dtype object.

Returns

NumPy type object.

Return type

Any

pytorch_pfn_extras.get_xp pytorch_pfn_extras.from_numpy_dtype