Namespace convert

Index

Type aliases

Backend

Functions

Type aliases

Backend

Backend: "CPU" | "WASM" | "GPU"

Functions

sameType

sameType<DTpe1, DTpe2>(a: Tensor<DTpe1>, b: Tensor<DTpe2>): boolean

- Defined in util/convert.ts:116
Determines if the two tensors are of the same type, ie. if they are on the same backend

Type parameters
- DTpe1: DType
- DTpe2: DType
Parameters
- a: Tensor<DTpe1>
- b: Tensor<DTpe2>
Returns boolean

toBackend

toBackend<DTpe>(tensor: Tensor<DTpe>, backend: Backend): Promise<Tensor<DTpe>>

- Defined in util/convert.ts:13
Convert a tensor to the specified backend

Type parameters
- DTpe: DType
Parameters
- tensor: Tensor<DTpe>
- backend: Backend
Returns Promise<Tensor<DTpe>>

toCPU

toCPU<DTpe>(tensor: Tensor<DTpe>): Promise<Tensor<DTpe>>

- Defined in util/convert.ts:26
Type parameters
- DTpe: DType
Parameters
- tensor: Tensor<DTpe>
Returns Promise<Tensor<DTpe>>

toGPU

toGPU<DTpe>(tensor: Tensor<DTpe>): Promise<Tensor<DTpe>>

- Defined in util/convert.ts:81
Type parameters
- DTpe: DType
Parameters
- tensor: Tensor<DTpe>
Returns Promise<Tensor<DTpe>>

toWASM

toWASM<DTpe>(tensor: Tensor<DTpe>): Promise<Tensor<DTpe>>

- Defined in util/convert.ts:48
Type parameters
- DTpe: DType
Parameters
- tensor: Tensor<DTpe>
Returns Promise<Tensor<DTpe>>

Generated using TypeDoc