torch.ReductionKernelCPU
export interface ReductionKernelCPU<DT extends DType> {
/**
* Execute the reduction operation.
*
* @param input - Input array
* @param out - Output array (pre-allocated)
* @param numel - Number of elements in input
* @param dim - Dimension to reduce (null for global reduction)
* @param keepdim - Whether to keep the reduced dimension
*/
forward(
input: TypedArrayFor<DT>,
out: TypedArrayFor<DT>,
numel: number,
dim: number | null,
keepdim: boolean
): void;
}DTextends DTypeforward((input: TypedArrayFor<DT>, out: TypedArrayFor<DT>, numel: number, dim: number | null, keepdim: boolean) => void)- – Execute the reduction operation.
CPU kernel signature for reduction operations.