@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
12 lines (11 loc) • 418 B
TypeScript
import { DType, PadMode, Tensor } from '../../../library';
import { BackwardOp, VariableI } from '../../types';
export declare class PadBack<DTpe extends DType> implements BackwardOp<DTpe> {
x: VariableI<DTpe>;
pads: number[];
mode: PadMode;
value: number;
constructor(x: VariableI<DTpe>, pads: number[], mode: PadMode, value: number);
backward(grad: Tensor<DTpe>): void;
delete(): void;
}