FillOperation

功能

将指定位置设置为value值或者生成一个指定Shape的Tensor并填充为value。

图1 FillOperation

功能实现

Fill 算子的功能是将指定位置设置为 value 值或者生成一个指定 Shape 的 Tensor 并填充为 value。

• MaskedFill 功能（withMask 为 true）：

  若 x 里的某元素，其所对应 mask 里同位置元素值为 true 或 1，则 output 该位置元素被设置为 value 值，output 其余位置元素值与 x 相同。支持广播功能。

  图2 Fill算子实现原理
  
  图3 Fill算子实现原理（广播场景）
  

• Fill 功能（withMask 为 false）：

  生成一个由 outDim 指定 shape 的 tensor，所有元素数值均为 value。

定义

struct FillParam {
    bool withMask = true;
    SVector<float> value;
    SVector<int64_t> outDim;
}

参数列表

• withMask为true

  将由 mask 指定的位置的元素设置为 value 值。

  inTensors 数量为 2，不需要指定 outDim。

• withMask为false

  生成一个由 outDim 指定 Shape 的 Tensor 并将其每个元素填充为 value。

  inTensors 数量为 0，需指定 outDim。

输入

输出

规格约束

• Param参数约束

  a. value.size() == 1。

• 输入输出约束
  • 输入x不可以被broadcast。
  • 输入mask的每个元素只能是0或者1。
  • withMask 为 true 时，mask shape 与 x 一致或可以 broadcast 为 x 的shape。

接口调用示例

• 示例1：MaskedFill 非广播场景
  • 输入

    param.withMask = true
    param.value = [-1]
    x = [[1, 2, 3],
         [4, 5, 6],
         [7, 8, 9]]
    mask = [[ true, false,  true],
             [false,  true,  true],
             [ true,  true, false]]

  • 输出

    output = [[-1,  2, -1],
               [ 4, -1, -1],
               [-1, -1,  9]]

• 示例2：MaskedFill 广播场景
  • 输入

    param.withMask = true
    param.value = [-1]
    x = [[1, 2, 3],
         [4, 5, 6],
         [7, 8, 9]]
    mask = [true, false, true]

  • 输出

    output = [[-1,  2, -1],
               [-1,  5, -1],
               [-1,  8, -1]]

• 示例3：Fill 场景
  • 输入

    param.withMask = false
    param.value = [1]
    param.outDim = [2, 3]

  • 输出

    output = [[1, 1, 1],
               [1, 1, 1]]