rml_op_slice_params

Parameters for the RML_OP_SLICE operation.

Semantic is the same as for the start:stop:step notation in Numpy. Additionally, ‘axes’ specifies axes to apply slice. See details.

API

Graph Manipulation

Items

Name

Type

Description

input

rml_op

Operation that outputs tensor of data to extract slices from.

axes

rml_op

Operaton that outputs 1-D tensor of axes that starts and ends apply to.
Negative value means counting dimensions from the back.
Accepted range is [-R, R-1] where R = rank(data).

starts

rml_op

Operaton that outputs 1-D tensor of starting indices of corresponding axis in axes.

ends

rml_op

Operaton that outputs 1-D tensor of ending indices (exclusive) of corresponding axis in axes.

strides

rml_op

Operation that outputs tensor of slice step of corresponding axis in axes.
Negative value means slicing backward. steps cannot be 0.
Only stride=1 is supported now