LinSpace

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品
Atlas 训练系列产品
Kirin X90 处理器系列产品
Kirin 9030 处理器系列产品

功能说明

  • 算子功能:生成一个等间隔数值序列。创建一个大小为num的1维向量,其值从start起始到stop结束(包含)线性均匀分布。

  • 计算公式:

output=(start,start+stop−startnum−1,...,start+(num−2)∗stop−startnum−1,stop)output = (start, start + \frac{stop - start}{num - 1},...,start + (num - 2) * \frac{stop - start}{num -1}, stop)

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
start 输入 公式中的输入张量start,必须为1维张量。 INT8、UINT8、INT16、INT32、FLOAT、FLOAT16、BFLOAT16 ND
stop 输入 公式中的输入张量stop,必须为1维张量。 INT8、UINT8、INT16、INT32、FLOAT、FLOAT16、BFLOAT16 ND
num 输入 公式中的输入张量num,必须为1维张量。 INT32、INT64 ND
output 输出 公式中的输出张量output。 INT8、UINT8、INT16、INT32、FLOAT、FLOAT16、BFLOAT16 ND
  • Kirin X90/Kirin 9030 处理器系列产品、Atlas 训练系列产品、Atlas 推理系列产品:不支持BFLOAT16、INT16。

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_lin_space 通过aclnnLinspace接口方式调用LinSpace算子。
图模式调用 test_geir_lin_space 通过算子IR构图方式调用LinSpace算子。