Class ComputationalGraph

constructor

• new ComputationalGraph(): ComputationalGraph

  Returns ComputationalGraph

_nodes

_order

inputNodes

• get inputNodes(): Node[]

  Input nodes

  Returns Node[]

nodes

• get nodes(): Node[]

  Graph nodes

  Returns Node[]

outputNodes

• get outputNodes(): Node[]

  Output nodes

  Returns Node[]

size

• get size(): number

  Number of nodes

  Returns number

_calcOrder

• _calcOrder(): void

  Returns void

add

• add(
      layer: PlainLayerObject | Layer,
      name?: string,
      inputs?: string | number | (string | number)[],
  ): void

  Add a layer.

  Parameters

  • layer: PlainLayerObject | Layer

    Added layer
  • Optionalname: string

    Node name
  • Optionalinputs: string | number | (string | number)[]

    Input node names or const value for the added layer

  Returns void

bind

• bind(values: object): void

  Bind values to layers

  Parameters

  • values: object

    Binding values

  Returns void

calc

• calc(require?: (string | number)[]): void

  Returns calculated values.

  Parameters

  • Optionalrequire: (string | number)[]

    Name or index of nodes at least calculated

  Returns void

getNode

• getNode(name: string): Node

  Returns a specific name node.

  Parameters

  • name: string

    Node name

  Returns Node

  Node

grad

• grad(e?: Matrix): Matrix

  Returns gradient values.

  Parameters

  • Optionale: Matrix

    Input of gradient

  Returns Matrix

  Output of gradient

toDot

• toDot(): string

  Returns a string of DOT format.

  Returns string

  String of DOT format

toObject

• toObject(): LayerObject[]

  Returns object representation.

  Returns LayerObject[]

  Object represented this graph

toONNX

• toONNX(): Uint8Array

  Returns onnx model

  Returns Uint8Array

  onnx model byte array

StaticfromObject

• fromObject(nodes: LayerObject[]): ComputationalGraph

  Returns Graph.

  Parameters

  • nodes: LayerObject[]

    Array of object represented a graph

  Returns ComputationalGraph

  Graph

StaticfromONNX

• fromONNX(
      buffer: ArrayBuffer | Uint8Array<ArrayBufferLike> | File,
  ): Promise<ComputationalGraph>

  Load onnx model.

  Parameters

  • buffer: ArrayBuffer | Uint8Array<ArrayBufferLike> | File

    File

  Returns Promise<ComputationalGraph>

  Loaded graph