initializers

Documentation

Overview

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

Index

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

Functions

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(stddev float64) VariableInitializer

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

func RandomUniformFn

func RandomUniformFn(min, max float64) VariableInitializer

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