GPUBackend | /packages/core/src/backends/gpu/backend.ts | netsaur@0.4.2

class GPUBackend

implements Backend

import { GPUBackend } from "https://deno.land/x/netsaur@0.4.2/packages/core/src/backends/gpu/backend.ts";

GPU Backend.

Constructors

new

GPUBackend(

library: Library,

outputShape: Shape<Rank>,

id: bigint,

)

library: Library

outputShape: Shape<Rank>

layers: number[],

outputShape: Shape<Rank>,

): Promise<Tensor<Rank>>

save(): Uint8Array

saveFile(path: string): void

datasets: DataSet[],

epochs: number,

batches: number,

rate: number,

): void

load(buffer: Uint8Array, library: Library): GPUBackend

loadFile(path: string, library: Library): GPUBackend