Select

Slices the input tensor at the dimension and index specified.

Takes two inputs, input and index, and removes the dimension <Dimension> from the input by only including the value at index in the output.

For the input index, the data type must be Int8, Int16, Int32, Int64 or Uint8, and it must be a scalar. That is, it must have the shape (). For the input input, any data type and shape is supported. The output shape will be the same as the input shape, but with <Dimension> removed.

For example, given an input with shape (2, 3, 4), and <Dimension> set to 1, the output shape will be (2, 4). For another example, given an input with shape (3, 3), <Dimension> set to 0, value for input equal to the tensor [[1, 2, 3], [4, 5, 6], [7, 8, 9]], and value for index equal to the scalar 2, the output will be a tensor of shape (3) and with value [7, 8, 9]. This is because we selected the element at index 2 of the 0th dimension.

You can choose a value for <Dimension> by clicking the node and configuring it on the right panel.