rml_op_unsqueeze_params
Parameters for the RML_OP_UNSQUEEZE operation.
Inserts a dimension of size 1 at the dimension index axis of input’s shape.
API
Graph Manipulation
Items
Name |
Type |
Description |
|---|---|---|
input |
Operation that outputs tensor containing input data. |
|
num_axes |
size_t |
Number of axes to unsqueeze. |
axes[RML_TENSOR_MAX_RANK] |
int32_t |
List of integers indicating the dimensions to unsqueeze. Negative value means counting dimensions from the back. Accepted range is [-R, R-1] where R = rank(data). |