@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
9 lines (8 loc) • 392 B
TypeScript
import { DType, Tensor } from '../../../library';
import { BackwardOp, VariableI } from '../../types';
export declare class AveragePoolBack<DTpe extends DType> implements BackwardOp<DTpe> {
x: VariableI<DTpe>;
constructor(x: VariableI<DTpe>, kernelShape: number[], pads: number[], strides: number[], includePad: boolean);
backward(grad: Tensor<DTpe>): void;
delete(): void;
}