接口 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 生成