Class ReductionBroadcaster

Broadcasting utilities for reduction operations (sum, mean, etc.)

Index

Constructors

Methods

getReducedShape reduce

Constructors

constructor

new ReductionBroadcaster(): ReductionBroadcaster
Returns ReductionBroadcaster

Methods

`Static`getReducedShape

getReducedShape(inputShape: Shape, axes?: number[], keepDims?: boolean): Shape
Compute the shape after reduction along specific axes
Parameters
- inputShape: Shape
- Optionalaxes: number[]
- keepDims: boolean = false
Returns Shape
- Defined in packages/core/src/shape/broadcasting.ts:570

`Static`reduce

reduce<T extends MutableArrayLike>(
    inputShape: Shape,
    input: ArrayLike<number>,
    operation: (accumulator: number, value: number) => number,
    initialValue: number,
    axes?: number[],
    keepDims?: boolean,
    outputConstructor?: new (length: number) => T,
): { result: T; shape: Shape }
Execute a reduction operation
Type Parameters
- T extends MutableArrayLike
Parameters
- inputShape: Shape
- input: ArrayLike<number>
- operation: (accumulator: number, value: number) => number
- initialValue: number
- Optionalaxes: number[]
- keepDims: boolean = false
- outputConstructor: new (length: number) => T = ...
Returns { result: T; shape: Shape }
- Defined in packages/core/src/shape/broadcasting.ts:598