接口 TensorFromMLTensorOptions<T>

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

类型参数

T extends MLTensorDataTypes

层级

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

索引

属性

方法

dispose? download?

属性

`可选` dataType

dataType?: T

描述张量的数据类型。

`只读` dims

dims: readonly number[]

获取张量的维度。

方法

`可选` dispose

dispose(): void
一个可选的回调函数，将在张量被释放时调用。

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

返回 void
继承自 GpuResourceConstructorParameters.dispose
- 定义于 tensor-factory.ts:39

`可选` download

download(): Promise<DataTypeMap[T]>
一个可选的回调函数，用于将数据从 GPU 下载到 CPU。

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

返回 Promise<DataTypeMap[T]>
继承自 GpuResourceConstructorParameters.download
- 定义于 tensor-factory.ts:32

使用 TypeDoc 生成