@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
15 lines (14 loc) • 562 B
TypeScript
import { DType, Tensor } from '../../../library';
import { BackwardOp, VariableI } from '../../types';
export declare class GemmBack<DTpe extends DType> implements BackwardOp<DTpe> {
a: VariableI<DTpe>;
b: VariableI<DTpe>;
transA: boolean;
transB: boolean;
alpha: number;
beta: number;
c?: VariableI<DTpe> | undefined;
constructor(a: VariableI<DTpe>, b: VariableI<DTpe>, transA: boolean, transB: boolean, alpha: number, beta: number, c?: VariableI<DTpe> | undefined);
backward(grad: Tensor<DTpe>): void;
delete(): void;
}