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 |
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. |