def parse_input_dtype(input_dtype: Any) -> _enums.dtype

函数功能

静态方法，解析输入input_type为_enums.dtype类型。

函数原型

def parse_input_dtype(input_dtype: Any) -> _enums.dtype

约束说明

如果input_dtype为_enums.dtype和torch.dtype之外的类型，则会无法解析并抛出异常。

参数说明

返回值说明

• 如果input_dtype为_enums.dtype类型，则直接返回该值。
• 如果input_dtype为torch.dtype类型，则转换成_enums.dtype类型并返回。