stack

Function stack 

Source
pub fn stack<T: Scalar>(tensors: &[Tensor<T>], dim: usize) -> Result<Tensor<T>>
Expand description

Stacks tensors along a new dimension.

ยงArguments

  • tensors - Slice of tensors to stack
  • dim - Dimension at which to insert the new axis