@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
15 lines (14 loc) • 584 B
TypeScript
import { DType, Tensor } from '../../../library';
import { BackwardOp, VariableI } from '../../types';
export declare class ConvBack<DTpe extends DType> implements BackwardOp<DTpe> {
x: VariableI<DTpe>;
w: VariableI<DTpe>;
strides: number[];
padding: number[];
dilations: number[];
group: number;
b?: VariableI<DTpe> | undefined;
constructor(x: VariableI<DTpe>, w: VariableI<DTpe>, strides: number[], padding: number[], dilations: number[], group: number, b?: VariableI<DTpe> | undefined);
backward(grad: Tensor<DTpe>): void;
delete(): void;
}