@tensorflow/tfjs-core/dist/public/chained_ops/split.d.ts

Hardware-accelerated JavaScript library for machine intelligence

import { Rank } from '../../types';
declare module '../../tensor' {
    interface Tensor<R extends Rank = Rank> {
        split<T extends Tensor>(numOrSizeSplits: number[] | number, axis?: number): T[];
    }
}