介面 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 生成