T_AII 008-2023 深度学习算法框架通用接口规范.docxVIP

ICS35.020

CCSL70

团体标

准

T/AII008—2023

深度学习算法框架通用接口规范

Generalinterfacespecificationfordeeplearningalgorithmframeworks

2023-12-26发布

2023-12-26实施

深圳市人工智能行业协会发布

I

T/AII008—2023

目次

前言 III

引言 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应用层 10

附录A（规范性）基础接口 12

A.1张量Tensor 12

A.2模块Module 13

附录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.1initialize接口参数列表 12

表A.2index接口参数列表 12

表A.3permute接口参数列表 13

II

T/AII008—2023

表A.4reshape接口参数列表 13

表A.5forward接口参数列表 14

表A.6get_parameters接口参数列表 14

表A.7get_modules接口参数列表 14

表A.8load_state_dict接口参数列表 14

表A.9backward接口参数列表 15

表B.1construct_dataset接口参数列表 16

表B.2get_sample接口参数列表 16

表B.3construct_sampler接口参数列表 17

表B.4iterate_sample接口参数列表 17

表B.5iterate_batch接口参数列表 17

表B.6construct_optimizer接口参数列表 18

表B.7update_learnable_parameter接口参数列表 18

表B.8zero_gradient接口参数列表 19

表B.9construct_trainer接口参数列表 19

表B.10train_one_batch接口参数列表 19

表B.11train_one_epoch接口参数列表 20

表B.12construct_inference接口参数列表 20

表B.13infer_one_sample接口参数列表 21

表B.14infer_samples接口参数列表 21

表B.15construct_evalvater接口参数列表 22

表B.16evalvate接口参数列表 22

表B.17construct_operator接口参数列表 22

表B.18operate接口参数列表 23

III

T/AII008—2023

前言

本文件按照GB/T1.1-2020《标准化工作导则第1部分：标准化文件的结构和起草规则》的规定起草。

本文件由深圳市人工智能行业协会（AII）提出并归口。

本文件起草单位：博瀚智能（深圳）有限公司、北京大学、深圳云天励飞技术股份有限公司、达闼机器人股份有限公司、深圳市酷开网络科技股份有限公司、麒麟合盛网络技术股份有限公司、软通智慧科技有限公司、上海致宇信息技术有限公司、深思考人工智能机器人科技（北京）有限公司、北京智境云创科技有限公司、深圳优地科技有限公司、深圳云译科技有限公司、中电金信软件（深圳）有限公司、深圳云知声信息技术有限公司