GPUBackend | /src/backends/gpu/backend.ts | netsaur@0.3.0

class GPUBackend

implements Backend

import { GPUBackend } from "https://deno.land/x/netsaur@0.3.0/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[keyof Shape],

): Promise<Tensor<Rank>>

save(): Uint8Array

saveFile(path: string): void

datasets: DataSet[],

epochs: number,

batches: number,

rate: number,

)

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

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