ScatterElementsUpdate¶

Versioned name : ScatterElementsUpdate-3

Category : Data movement operations

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>