Function prepareSplitSize

  • prepareSplitSize(x, numOrSizeSplits, axis?): number[]

  • Prepare the split size array. When the input is a number, the axis is evenly divided among the split size. When the input contains the negative value, the rest of the axis is allocated toward that.

    Parameters

    • x: Tensor<Rank> | TensorInfo
    • numOrSizeSplits: number | number[]
    • Optional axis: number

    Returns number[]

Settings

Member Visibility

Theme

Generated using TypeDoc