Skip to main content

dstack

Function dstack 

Source
pub fn dstack<T>(tensors: &[Tensor<T>]) -> Result<Tensor<T>, FerrotorchError>
where T: Float,
Expand description

dstack(tensors) — stack tensors depth-wise (along dim 2 after promoting each to ≥3-D).

Mirrors upstream aten/src/ATen/native/TensorShape.cpp:3544 Tensor dstack(TensorList tensors): atleast_3d each input then cat(_, 2). Autograd inherited from reshape/unsqueeze + cat.