Python3.6+支持指定参数类型和函数返回类型,比如,
def fun(x: int) -> int:
pass
基本数据类型:
举例说明
Tuple[int, int]:输入只包含两个整数的整数类型元组。
Tuple[int]:输入整数类型元组。
List[int]:可以输入整数类型列表。
Optional[int] / Union[int, None]:可以输入一个整数类型或None。
Union[int, Callable]:可以输入一个整数类型或回调函数。
Union[List[int], str]:可以输入一个整数类型列表或一个字符串。
Callable[..., torch.nn.Module]:表示函数可以接受任意数量的参数,但是最后一个参数必须是 torch.nn.Module
类型。
CustomClass:输入一个自定义类。