接口 TypedTensor<T>

表示用于馈送或从模型推理中获取的多维数组。

interface TypedTensor<T> {
    data: DataTypeMap[T];
    dims: readonly number[];
    gpuBuffer: GpuBufferTypeFallback;
    location: DataLocation;
    mlTensor: MLTensorTypeFallback;
    size: number;
    texture: WebGLTexture;
    type: T;
    dispose(): void;
    getData(releaseData?): Promise<DataTypeMap[T]>;
    reshape(dims): TypedTensor<T>;
    toDataURL(options?): string;
    toImageData(options?): ImageData;
}

类型参数

继承关系

  • TypedTensorBase<T>
  • TypedTensorUtils<T>
    • TypedTensor

属性

data dims gpuBuffer location mlTensor size texture type

方法

dispose getData reshape toDataURL toImageData

属性

Readonly data

data: DataTypeMap[T]

获取张量的缓冲区数据。

如果数据不在 CPU 上（例如，它是 WebGL 纹理或 WebGPU 缓冲区的形式），则抛出错误。

Readonly dims

dims: readonly number[]

获取张量的维度。

Readonly gpuBuffer

gpuBuffer: GpuBufferTypeFallback

获取保存张量数据的 WebGPU 缓冲区。

如果数据不在 GPU 上且不是 WebGPU 缓冲区的形式，则抛出错误。

Readonly location

location: DataLocation

获取数据的位置。

Readonly mlTensor

mlTensor: MLTensorTypeFallback

获取保存张量数据的 WebNN MLTensor。

如果数据不在 WebNN MLTensor 中，则抛出错误。

Readonly size

size: number

获取张量中的元素数量。

Readonly texture

texture: WebGLTexture

获取保存张量数据的 WebGL 纹理。

如果数据不在 GPU 上且不是 WebGL 纹理的形式，则抛出错误。

Readonly type

type: T

获取张量的数据类型。

方法

dispose

  • dispose(): void

  • 释放张量数据。

    如果数据在 CPU 上，则移除其对底层数据的内部引用。如果数据在 GPU 上，则释放 GPU 上的数据。

    调用此函数后，该张量将被视为不再有效。其位置将被设置为 'none'。

    返回 void

getData

  • getData(releaseData?): Promise<DataTypeMap[T]>

  • 获取张量的缓冲区数据。

    如果数据在 CPU 上，则立即返回数据。如果数据在 GPU 上，则下载数据并返回 Promise。

    参数

    • Optional releaseData: boolean

      是否释放 GPU 上的数据。如果数据已在 CPU 上，则忽略此参数。

    返回 Promise<DataTypeMap[T]>

reshape

  • reshape(dims): TypedTensor<T>

  • 创建一个具有相同数据缓冲区和指定维度的新张量。

    参数

    • dims: readonly number[]

      新的维度。大小应与旧维度匹配。

    返回 TypedTensor<T>

toDataURL

  • toDataURL(options?): string

  • 从张量创建 DataURL 实例

    参数

    • Optional options: TensorToDataUrlOptions

      一个可选对象，表示从张量创建 DataURL 实例的选项。

      将应用以下默认设置：

      • format: 'RGB'
      • tensorLayout: 'NCHW'

    返回 string

    一个 DataURL 字符串，表示从张量数据转换的图像

toImageData

  • toImageData(options?): ImageData

  • 从张量创建 ImageData 实例

    参数

    • Optional options: TensorToImageDataOptions

      一个可选对象，表示从张量创建 ImageData 实例的选项。

      将应用以下默认设置：

      • format: 'RGB'
      • tensorLayout: 'NCHW'

    返回 ImageData

    一个 ImageData 实例，表示从张量数据转换的图像

设置

成员可见性

主题

当前页

使用 TypeDoc 生成