接口 MLTensorConstructorParameters<T>

interface MLTensorConstructorParameters<T> {
    dims: readonly number[];
    location: "ml-tensor";
    mlTensor: MLTensorTypeFallback;
    type: T;
    dispose?(): void;
    download?(): Promise<DataTypeMap[T]>;
}

类型参数

继承关系

  • CommonConstructorParameters<T>
  • GpuResourceConstructorParameters<T>
    • MLTensorConstructorParameters

属性

dims location mlTensor type

方法

dispose? download?

属性

只读 dims

dims: readonly number[]

获取张量的维度。

只读 location

location: "ml-tensor"

指定数据位置为“ml-tensor”。

只读 mlTensor

mlTensor: MLTensorTypeFallback

指定保存张量数据的 WebNN MLTensor。

只读 type

type: T

指定张量的数据类型。

方法

可选 dispose

  • dispose(): void

  • 一个可选的回调函数，将在张量被释放时调用。

    如果不提供，张量会将 GPU 数据视为外部资源。

    返回 void

可选 download

  • download(): Promise<DataTypeMap[T]>

  • 一个可选的回调函数，用于将数据从 GPU 下载到 CPU。

    如果不提供，张量会将 GPU 数据视为外部资源。

    返回 Promise<DataTypeMap[T]>

设置

成员可见性

主题

本页内容

使用 TypeDoc 生成