@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
10 lines (9 loc) • 364 B
TypeScript
import { DType, Tensor } from '../../../library';
import { BackwardOp, VariableI } from '../../types';
export declare class ExpandBack<DTpe extends DType> implements BackwardOp<DTpe> {
a: VariableI<DTpe>;
shape: readonly number[];
constructor(a: VariableI<DTpe>, shape: readonly number[]);
backward(grad: Tensor<DTpe>): void;
delete(): void;
}