ScatterElementsUpdate

Versioned name : ScatterElementsUpdate-3

Category : Data movement

Short description : Creates a copy of the first input tensor with updated elements specified with second and third input tensors.

Detailed description : For each entry in updates, the target index in data is obtained by combining the corresponding entry in indices with the index of the entry itself: the index-value for dimension equal to axis is obtained from the value of the corresponding entry in indices and the index-value for dimension not equal to axis is obtained from the index of the entry itself.

For instance, in a 3D tensor case, the update corresponding to the [i][j][k] entry is performed as below:

output[indices[i][j][k]][j][k] = updates[i][j][k] if axis = 0,
output[i][indices[i][j][k]][k] = updates[i][j][k] if axis = 1,
output[i][j][indices[i][j][k]] = updates[i][j][k] if axis = 2

update tensor dimensions are less or equal to the corresponding data tensor dimensions.

Attributes : ScatterElementsUpdate does not have attributes.

Inputs :

  • 1 : data tensor of arbitrary rank r and of type T. Required.

  • 2 : indices tensor with indices of type T_IND. The rank of the tensor is equal to the rank of data tensor. All index values are expected to be within bounds [0, s - 1] along axis of size s. If multiple indices point to the same output location then the order of updating the values is undefined. If an index points to non-existing output tensor element or is negative then exception is raised. Required.

  • 3 : updates tensor of shape equal to the shape of indices tensor and of type T. Required.

  • 4 : axis tensor with scalar or 1D tensor with one element of type T_AXIS specifying axis for scatter. The value can be in range [-r, r - 1] where r is the rank of data. Required.

Outputs :

  • 1 : tensor with shape equal to data tensor of the type T.

Types

  • T : any numeric type.

  • T_IND : any integer numeric type.

  • T_AXIS : any integer numeric type.

Example

<layer ... type="ScatterElementsUpdate">
    <input>
        <port id="0">
            <dim>1000</dim>
            <dim>256</dim>
            <dim>7</dim>
            <dim>7</dim>
        </port>
        <port id="1">
            <dim>125</dim>
            <dim>20</dim>
            <dim>7</dim>
            <dim>6</dim>
        </port>
        <port id="2">
            <dim>125</dim>
            <dim>20</dim>
            <dim>7</dim>
            <dim>6</dim>
        </port>
        <port id="3">     <!-- value [0] -->
            <dim>1</dim>
        </port>
    </input>
    <output>
        <port id="4" precision="FP32">
            <dim>1000</dim>
            <dim>256</dim>
            <dim>7</dim>
            <dim>7</dim>
        </port>
    </output>
</layer>