onnxruntime
1// Copyright (c) Microsoft Corporation. All rights reserved.
2// Licensed under the MIT License.
3
4import { Tensor } from '../../../tensor';
5import { WebGLInferenceHandler } from '../inference-handler';
6
7export const shape = (_inferenceHandler: WebGLInferenceHandler, inputs: Tensor[]): Tensor[] => {
8validateInputs(inputs);
9return [new Tensor([inputs[0].dims.length], 'int32', undefined, undefined, new Int32Array(inputs[0].dims))];
10};
11
12const validateInputs = (inputs: Tensor[]): void => {
13if (!inputs || inputs.length !== 1) {
14throw new Error('Shape requires 1 input.');
15}
16};
17