介面 TensorFromMLTensorOptions<T>

interface TensorFromMLTensorOptions<T> {
    dataType?: T;
    dims: readonly number[];
    dispose?(): void;
    download?(): Promise<DataTypeMap[T]>;
}

型別引數

繼承關係

  • Pick<Tensor, "dims">
  • GpuResourceConstructorParameters<T>
    • TensorFromMLTensorOptions

屬性

dataType?: T

描述張量的資料型別。

dims: readonly number[]

獲取張量的維度。

方法

  • 張量被釋放時將呼叫的可選回撥函式。

    如果未提供,張量會將 GPU 資料視為外部資源。

    返回 void

  • 從 GPU 下載資料到 CPU 的可選回撥函式。

    如果未提供,張量會將 GPU 資料視為外部資源。

    返回 Promise<DataTypeMap[T]>

使用 TypeDoc 生成