rml_op_stack_params
Parameters for the RML_OP_STACK operation.
RML_OP_STACK operation Stacks a list of tensors with rank R into single tensor with rank R+1. All tensors MUST have the same shape and type.
API
Graph Manipulation
Items
Name |
Type |
Description |
|---|---|---|
num_inputs |
size_t |
The number of input operations. |
inputs |
A list of operations that output tensors containing input data. |
|
axis |
int |
The axis to stack along. A negative value means counting dimensions from back. |