torch.UnaryKernelCPU

export interface UnaryKernelCPU<DT extends DType> {
  /**
   * Execute the unary operation.
   *
   * @param input - Input array
   * @param out - Output array (pre-allocated)
   * @param numel - Number of elements
   */
  forward(input: TypedArrayFor<DT>, out: TypedArrayFor<DT>, numel: number): void;
}

DTextends DType

forward((input: TypedArrayFor<DT>, out: TypedArrayFor<DT>, numel: number) => void): – Execute the unary operation.

CPU kernel signature for unary operations.