@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
13 lines (12 loc) • 455 B
TypeScript
import Tensor, { DType } from '../../../types';
import { BackwardOp, VariableI } from '../../types';
export declare class SliceBack<DTpe extends DType> implements BackwardOp<DTpe> {
a: VariableI<DTpe>;
starts: number[];
ends: number[];
axes: number[];
steps: number[];
constructor(a: VariableI<DTpe>, starts: number[], ends: number[], axes: number[], steps: number[]);
backward(grad: Tensor<DTpe>): void;
delete(): void;
}