torchtyping
一种适用于PyTorch的工具,用于对张量的形状和数据类型进行类型注解,通过编程检查确保张量规范,减少错误。支持多种注解,具有高度的可扩展性,包括形状、数据类型和维度名称。与typeguard集成,可进行运行时类型检查,提升代码的可读性和健壮性。