Python指定参数类型

Python3.6+支持指定参数类型和函数返回类型，比如，

def fun(x: int) -> int:
    pass

基本数据类型：

• int, long, float
• bool, str
• List, Tuple, Dict, Set
• Iterable, Iterator, Generator
• Callable
• Any

举例说明

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：输入一个自定义类。