@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
11 lines (10 loc) • 403 B
TypeScript
import { BackwardOp, VariableI } from '../../../../autograd/types';
import { Tensor } from '../../../../library';
import { DType } from '../../../../types';
export declare class BCEBack<DTpe extends DType> implements BackwardOp<DTpe> {
x: VariableI<DTpe>;
y: VariableI<DTpe>;
constructor(x: VariableI<DTpe>, y: VariableI<DTpe>);
backward(grad: Tensor<DTpe>): void;
delete(): void;
}