接口 TensorFromGpuBufferOptions<T>

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

类型参数

层级

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

属性

dataType? dims

方法

dispose? download?

属性

可选的 dataType

dataType?: T

描述张量的数据类型。

只读 dims

dims: readonly number[]

获取张量的维度。

方法

可选的 dispose

  • dispose(): void

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

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

    返回值 void

可选的 download

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

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

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

    返回值 Promise<DataTypeMap[T]>

设置

成员可见性

主题

在本页

使用 TypeDoc 生成