Function stack_padded 

pub fn stack_padded(rows: &[Tensor]) -> Result<Tensor>

Right-pad a set of (1, rows_i, d) tensors with zeros to a common row count and concatenate them into (B, rows_max, d).