pytorch_pfn_extras.from_numpy_dtype

pytorch_pfn_extras.from_numpy_dtype(numpy_dtype)

Returns PyTorch dtype for the given NumPy dtype.

Parameters

numpy_dtype (Any) – NumPy’s dtype object.

Returns

PyTorch type object.

Return type

torch.dtype

pytorch_pfn_extras.as_numpy_dtype pytorch_pfn_extras.cuda.stream