pub fn flatten_pair<B: Backend>(
y: <B as BackendTypes>::FloatTensorPrimitive,
final_state: <B as BackendTypes>::FloatTensorPrimitive,
) -> (<B as BackendTypes>::FloatTensorPrimitive, usize, usize)Expand description
Flatten the two outputs (y and final_state) and concatenate them along a
fresh axis-0 into a single 1-D tensor. Returns the combined tensor and the
per-output flat lengths needed to split it later.