rml_op_squeeze_params

Parameters for the RML_OP_SQUEEZE operation.

RML_OP_SQUEEZE operation removes single-dimensional entries from the shape of a tensor.

API

Graph Manipulation

Items

Name

Type

Description

input

rml_op*

Operation that outputs tensor containing the input data.

num_axes

size_t

Number of axes to squeeze.

axes[RML_TENSOR_MAX_RANK]

const int32_t*

List of integers indicating the dimensions to squeeze.
Negative value means counting dimensions from the back.
Accepted range is [-R, R-1] where R = rank(data).