import { TypedTensorUtils } from './tensor-utils'; /** * represent a basic tensor with specified dimensions and data type. */ interface TypedTensorBase { /** * Get the dimensions of the tensor. */ readonly dims: readonly number[]; /** * Get the data type of the tensor. */ readonly type: T; /** * Get the buffer data of the tensor. */ readonly data: Tensor.DataTypeMap[T]; } export declare namespace Tensor { interface DataTypeMap { float32: Float32Array; uint8: Uint8Array; int8: Int8Array; uint16: Uint16Array; int16: Int16Array; int32: Int32Array; int64: BigInt64Array; string: string[]; bool: Uint8Array; float16: never; float64: Float64Array; uint32: Uint32Array; uint64: BigUint64Array; } interface ElementTypeMap { float32: number; uint8: number; int8: number; uint16: number; int16: number; int32: number; int64: bigint; string: string; bool: boolean; float16: never; float64: number; uint32: number; uint64: bigint; } type DataType = DataTypeMap[Type]; type ElementType = ElementTypeMap[Type]; /** * represent the data type of a tensor */ type Type = keyof DataTypeMap; } /** * Represent multi-dimensional arrays to feed to or fetch from model inferencing. */ export interface TypedTensor extends TypedTensorBase, TypedTensorUtils { } /** * Represent multi-dimensional arrays to feed to or fetch from model inferencing. */ export interface Tensor extends TypedTensorBase, TypedTensorUtils { } export interface TensorConstructor { /** * Construct a new string tensor object from the given type, data and dims. * * @param type - Specify the element type. * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (type: 'string', data: Tensor.DataTypeMap['string'] | readonly string[], dims?: readonly number[]): TypedTensor<'string'>; /** * Construct a new bool tensor object from the given type, data and dims. * * @param type - Specify the element type. * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (type: 'bool', data: Tensor.DataTypeMap['bool'] | readonly boolean[], dims?: readonly number[]): TypedTensor<'bool'>; /** * Construct a new numeric tensor object from the given type, data and dims. * * @param type - Specify the element type. * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new >(type: T, data: Tensor.DataTypeMap[T] | readonly number[], dims?: readonly number[]): TypedTensor; /** * Construct a new float32 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Float32Array, dims?: readonly number[]): TypedTensor<'float32'>; /** * Construct a new int8 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Int8Array, dims?: readonly number[]): TypedTensor<'int8'>; /** * Construct a new uint8 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Uint8Array, dims?: readonly number[]): TypedTensor<'uint8'>; /** * Construct a new uint16 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Uint16Array, dims?: readonly number[]): TypedTensor<'uint16'>; /** * Construct a new int16 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Int16Array, dims?: readonly number[]): TypedTensor<'int16'>; /** * Construct a new int32 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Int32Array, dims?: readonly number[]): TypedTensor<'int32'>; /** * Construct a new int64 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: BigInt64Array, dims?: readonly number[]): TypedTensor<'int64'>; /** * Construct a new string tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: readonly string[], dims?: readonly number[]): TypedTensor<'string'>; /** * Construct a new bool tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: readonly boolean[], dims?: readonly number[]): TypedTensor<'bool'>; /** * Construct a new float64 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Float64Array, dims?: readonly number[]): TypedTensor<'float64'>; /** * Construct a new uint32 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Uint32Array, dims?: readonly number[]): TypedTensor<'uint32'>; /** * Construct a new uint64 tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: BigUint64Array, dims?: readonly number[]): TypedTensor<'uint64'>; /** * Construct a new tensor object from the given type, data and dims. * * @param type - Specify the element type. * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (type: Tensor.Type, data: Tensor.DataType | readonly number[] | readonly boolean[], dims?: readonly number[]): Tensor; /** * Construct a new tensor object from the given data and dims. * * @param data - Specify the tensor data * @param dims - Specify the dimension of the tensor. If omitted, a 1-D tensor is assumed. */ new (data: Tensor.DataType, dims?: readonly number[]): Tensor; } export declare const Tensor: TensorConstructor; export {};