Interface TensorFromMLTensorOptions<T>

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

Type Parameters

Hierarchy

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

Properties

dataType? dims

Methods

dispose? download?

Properties

Optional dataType

dataType?: T

Describes the data type of the tensor.

Readonly dims

dims: readonly number[]

Get the dimensions of the tensor.

Methods

Optional dispose

  • dispose(): void

  • an optional callback function that will be called when the tensor is disposed.

    If not provided, the tensor treat the GPU data as external resource.

    Returns void

Optional download

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

  • an optional callback function to download data from GPU to CPU.

    If not provided, the tensor treat the GPU data as external resource.

    Returns Promise<DataTypeMap[T]>

Settings

Member Visibility

Theme

On This Page

Generated using TypeDoc