initializers

Documentation

Overview

Package initializers include several weight initializers, to be used with context. They implement computation.VariableInitializer type.

Index

• Constants
• func Finalize()
• func One(graph *Graph, shape shapes.Shape) *Node
• func Zero(graph *Graph, shape shapes.Shape) *Node
• type VariableInitializer
  • func RandomNormalFn(initialSeed int64, stddev float64) VariableInitializer
  • func RandomUniformFn(initialSeed int64, min, max float64) VariableInitializer

Constants

const NoSeed = int64(0)

Functions

func Finalize

func Finalize()

Finalize will clear the global state kept alive and free up the memory. Namely, the random number generator states.

Used for testing and debugging.

func One

func One(graph *Graph, shape shapes.Shape) *Node

One initializes variables with one.

func Zero

func Zero(graph *Graph, shape shapes.Shape) *Node

Zero initializes variables with zero.

Types

type VariableInitializer

type VariableInitializer func(graph *Graph, shape shapes.Shape) *Node

VariableInitializer builds a node that returns a value to initialize a variable of the given shape. It is defined in the Context.

func RandomNormalFn

func RandomNormalFn(initialSeed int64, stddev float64) VariableInitializer

RandomNormalFn returns an initializer that generates random normal values with the given standard deviation and mean set to 0.

The parameter `initialSeed` is used to initialize the random number generator -- only the first time it is used for a graph. If it is set to 0 (NoSeed), a random seed is instead generated (from the nanosecond clock).

func RandomUniformFn

func RandomUniformFn(initialSeed int64, min, max float64) VariableInitializer

RandomUniformFn return an initializer that generates a random uniform values from [min, max).

The parameter `initialSeed` is used to initialize the random number generator -- only the first time it is used for a graph. If it is set to 0 (NoSeed), a random seed is instead generated (from the nanosecond clock).