@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
12 lines (11 loc) • 451 B
TypeScript
import { DType, Tensor } from '../../../library';
import { BackwardOp, VariableI } from '../../types';
export declare class ProductBack<DTpe extends DType> implements BackwardOp<DTpe> {
input: VariableI<DTpe>;
product: Tensor<DTpe>;
sumDims: number[];
keepDims: boolean;
constructor(input: VariableI<DTpe>, product: Tensor<DTpe>, sumDims: number[], keepDims: boolean);
backward(grad: Tensor<DTpe>): void;
delete(): void;
}