Function split_

tfjs-core/src/ops/split.ts:62–72

* Splits a `tf.Tensor` into sub tensors. * * If `numOrSizeSplits` is a number, splits `x` along dimension `axis` * into `numOrSizeSplits` smaller tensors. * Requires that `numOrSizeSplits` evenly divides `x.shape[axis]`. * * If `numOrSizeSplits` is a number array, splits `x` into * `numOrSize

(
    x: Tensor|TensorLike, numOrSizeSplits: number[]|number, axis = 0)

 * @doc {heading: 'Tensors', subheading: 'Slicing and Joining'}
 */
function split_<T extends Tensor>(
    x: Tensor|TensorLike, numOrSizeSplits: number[]|number, axis = 0): T[] {
  const $x = convertToTensor(x, 'x', 'split');

  const inputs: SplitVInputs = {x: $x};
  const attr: SplitVAttrs = {numOrSizeSplits, axis};

  return ENGINE.runKernel(
             SplitV, inputs as unknown as NamedTensorMap,
             attr as unknown as NamedAttrMap) as unknown as T[];
}

export const split = /* @__PURE__ */ op({split_});