torch.StackShape
export type StackShape<S extends Shape, N extends number, Dim extends number> = InsertDim<
S,
Dim,
N
>;Sextends ShapeNextends numberDimextends numberCompute output shape of stack operation. Stack inserts a new dimension of size N (number of tensors) at position Dim. StackShape<[2, 3], 3, 0> = [3, 2, 3] (stack 3 tensors of shape [2,3] at dim 0)