pub fn stack(tensors: Vec<PyTensor>, dim: i32) -> PyResult<PyTensor>
Stack tensors along a new dimension