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

ICS35.020CCSL70

团体标准

T/AII008—2023

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

Generalinterfacespecificationfordeeplearningalgorithmframeworks

26发布2023-12-2023-12-26实施

26发布

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

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

I

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接口参数列表 2

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

1

T/AII008—2023

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

1范围

本文件确立了面向深度学习算法框架通用应用接口模型、规定了接口功能及技术要求。

本文件适用于基于深度学习的算法开发和应用过程中所涉及应用编程接口的适配与调用。

2规范性引用文件

下列文件中的内容通过文中的规范性引用而构成本文件必不可少的条款。其中，注日期的引用文件，仅该日期对应的版本适用于本文件；不注日期的引用文件，其最新版本(包括所有的修改单)适用于本文件。

IS0/IEC22989:2022信息技术人工智能概念和术语(Informationtechnology—Artificial

i