FilesLast commitLast update
1 month ago
1 month ago
1 month ago
1 month ago
1 month ago
1 month ago
README

QueueData

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT √
Atlas A3 训练系列产品/Atlas A3 推理系列产品 √
Atlas A2 训练系列产品/Atlas A2 推理系列产品 √
Atlas 200I/500 A2 推理产品 √
Atlas 推理系列产品 √
Atlas 训练系列产品 √

功能说明

为其他算子提供队列数据。QueueData算子作为图中的队列数据输入节点,根据output_types和output_shapes属性计算输出张量的长度(每项含64字节信息头、维度描述及数据字节),输出为DT_UINT8类型的一维张量。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
index 属性 输入张量的索引。数据类型为int32或int64。假设网络中有三个data节点,一个应设为0,另一个设为1,第三个应设为2。 Int,默认值为0 -
queue_name 属性 队列名称。 String,默认值为"" -
output_types 属性 输出数据的数据类型列表。 ListType,默认值为{} -
output_shapes 属性 输出数据的形状列表。 ListListInt,默认值为{{}, {}} -
y 输出 DT_UINT8类型的一维张量,长度为根据output_types和output_shapes计算的总字节数。 DT_UINT8 ND

约束说明

  • output_types和output_shapes的长度必须一致。

调用说明

QueueData算子为图引擎基础算子,通常作为图中队列数据输入节点使用,不单独调用。

调用方式 调用样例 说明
图模式调用 test_geir_queue_data 通过算子IR构图方式调用QueueData算子