Function unstack

  • Unstacks a rank-R tensor into an array of rank-(R-1) tensors.

    Parameters

    • t: Tensor

      the input tensor

    • Optional axis: number

      (optional) the axis to unstack along

    Returns Tensor[]

    the array of tensors