GetDynamicDims

功能说明

GetDynamicDims算子用于根据shape_info中标记的未知维度,从动态输入张量中提取实际维度值并输出。

参数说明

参数名 输入/输出 参数类型 数据类型 说明
input 输入 Tensor动态输入 int32、int64 输入张量,每个输入表示一个shape向量。
dims 输出 Tensor int32、int64 输出未知维度的实际取值。
shape_info 属性 ListInt int64 各输入的shape信息,按rank, dim0, dim1, ...编码,-1表示未知维度。
N 属性 Int int64 动态输入个数。

约束说明

  • 输入input数量必须与属性N一致。
  • shape_info中描述的输入个数和每个输入rank必须与实际输入一致。
  • 输出dims的数据类型支持int32和int64。

调用说明

调用方式 样例链接 说明
图模式调用 test_geir_get_dynamic_dims 通过算子IR构图方式调用GetDynamicDims算子。