Function stack

  • Stacks an array of tensors along an axis. Tensors must have the same rank.

    Parameters

    • tensors: Tensor[]

      the tensors to be stacked

    • Optional axis: number

      (optional) the axis to stack along

    Returns Tensor

    the stacked tensor