ICS35.020 CCS L70 团 体 标 准 T/AIⅡ008—2023 深度学习算法框架通用接口规范 General interface specification for deep learning algorithm frameworks 2023-12-26发布 2023-12-26实施 深圳市人工智能行业协会发布 T/A1I008—2023 目 次 前言 ....III 引言 444 ...IV 1范围... 1 2规范性引用文件. 1 3术语和定义 . .1 4缩略语. 2 5深度学习算法框架通用接口模型 .3 5.1算法接口目标.. 3 5.2算法接口模型 3 6深度学习算法框架通用接口要求. 6 6.1框架适配层.. . 6 6.2核心算法层. ..6 6.3应用层. 附录A(规范性)基础接口. ...12 A.1张量Tensor... 12 A.2模块Module.... 附录B(规范性)接口参数表. .....16 B.1数据读取 ..........16 B.2采样器. ...16 B.3优化器..... .......18 B.4训练器.. .....19 B.5推理器. ....20 B.6评估器. ......21 B.7数据操作. .....22 参考文献 ............24 图1系统结构标准说明图 4 图2总体接口图 5 表A.I initialize接口参数列表 12 表A.2 index接口参数列表 12 表A.3 permute接口参数列表 13 I T/A11008—2023 表A.4 reshape接口参数列表 13 表A.5 forward接口参数列表 14 表A.6 get parameters接口参数列表 14 表A.7 get modules接口参数列表 14 表A.8 load state dict接口参数列表 14 表A.9 backward接口参数列表 15 表B.1 construct_dataset接口参数列表16 表B.2get_sample接口参数列表 .16 表B.3 construct_sampler接口参数列表 ...17 表B.4 iterate_sample接口参数列表17 表B.5 iterate batch接口参数列表. ..........17 表B.6 construct_optimizer接口参数列表 .18 表B.7 update_learnable_parameter接口参数列表18 表B.8 zero gradient接口参数列表 .19 表B.9 construct trainer接口参数列表. .....19 表B.10 train_one_batch接口参数列表19 表B.11 train_one_epoch接口参数列表 表B.12 construct inference接口参数列表 ..20 表B.13 infer_one_sample接口参数列表21 表B.14 infer_samples接口参数列表 表B.15 construct_evalvater接口参数列表. 22 表B.16 evalvate接口参数列表 22 表B.17 construct operator接口参数列表 22 表B.18 operate接口参数列表 .2...
T/AII 008-2023 深度学习算法框架通用接口规范.pdf
