Type Alias Sum<T, Axes, KeepDims>

Sum: SumOp<T, Axes, KeepDims>

Type-level sum function for use in other type computations

Type Parameters

T extends AnyTensorStorage
Input tensor storage type
Axes extends readonly number[] | undefined = undefined
Axes to reduce along
KeepDims extends boolean = false
Whether to keep reduced dimensions

Example

type Input = TensorStorage<Float32, [2, 3, 4], [12, 4, 1], DefaultLayoutFlags>;
type BatchSum = Sum<Input, [0]>; // Sum over batch dimension