Type Alias Mean<T, Axes, KeepDims>

Mean: MeanOp<T, Axes, KeepDims>

Type-level mean 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<Int32, [32, 128], [128, 1], DefaultLayoutFlags>;
type FeatureMean = Mean<Input, [-1], true>; // Mean over features, keep dim