@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
12 lines (11 loc) • 510 B
TypeScript
import { Tensor } from '../tensor';
export declare function validateUpdateShape(shape: number[], indices: Tensor, updates: Tensor): void;
export interface ScatterShapeInfo {
sliceRank: number;
numUpdates: number;
sliceSize: number;
strides: number[];
outputSize: number;
}
export declare function validateInput(updates: Tensor, indices: Tensor, shape: number[]): void;
export declare function calculateShapes(updates: Tensor, indices: Tensor, shape: number[]): ScatterShapeInfo;