范围¶

范围 - 11¶

版本¶

名称: Range (GitHub)
域名: main
since_version: 11
function: True
support_level: SupportType.COMMON
形状推断: True

此版本的运算符自版本11起可用。

摘要¶

生成一个包含从start开始并以delta为增量递增的数列的张量，直到limit（不包括）。

输出范围中的元素数量计算如下：

number_of_elements = max( ceil( (limit - start) / delta ) , 0 )

确定输出内容的伪代码如下所示：

for(int i=0; i<number_of_elements; ++i) {
  output[i] =  start + (i * delta);
}

示例 1

Inputs: start = 3, limit = 9, delta = 3
Output: [3, 6]

示例 2

Inputs: start = 10, limit = 4, delta = -2
Output: [10, 8, 6]

函数体¶

此运算符的函数定义。

<
  domain: "",
  opset_import: ["" : 11]
>
Range (start, limit, delta) => (output)
{
   sub_result = Sub (limit, start)
   sub_result_casted = Cast <to: int = 1> (sub_result)
   delta_casted = Cast <to: int = 1> (delta)
   div_result = Div (sub_result_casted, delta_casted)
   ceil_result = Ceil (div_result)
   ceil_result_relu = Relu (ceil_result)
   ceil_result_relu_int = Cast <to: int = 7> (ceil_result_relu)
   ceil_result_relu_bool = Cast <to: int = 9> (ceil_result_relu)
   variadic_output, output = Loop (ceil_result_relu_int, ceil_result_relu_bool, start) <body: graph = loop_body_attribute (int64 i, bool cond,  prev) => ( cond_out,  current,  range) {
      cond_out = Identity (cond)
      current = Add (prev, delta)
      range = Identity (prev)
   }>
}

输入¶

start（异构） - T:

标量。输出值范围的第一个条目。
limit (异构) - T:

标量。输出值范围的独占上限。
delta (异构) - T:

标量。步进的值。

输出¶

输出 (异构) - T:

一个与输入类型相同的1维张量，包含生成的值范围。

类型约束¶

T 在 ( tensor(double), tensor(float), tensor(int16), tensor(int32), tensor(int64) ):

将输入类型限制为常见的数值类型张量。