Add constant weight initializer

initwfn/Constant.go

@@ -46,3 +46,28 @@ func (o OnesConfig) Type() Type {
 func (o OnesConfig) Create() G.InitWFn {
 	return G.Ones()
 }
+
+// ConstantConfig implements a configuration of a weight initializer
+// that initializes all weights to a constant value.
+type ConstantConfig struct {
+	Value float64
+}
+
+// Constant returns a new zeroes weight intializer
+func NewConstant(value float64) (*InitWFn, error) {
+	config := ConstantConfig{value}
+
+	return newInitWFn(config)
+}
+
+// Type returns the type of the weight initializer created using this
+// config
+func (c ConstantConfig) Type() Type {
+	return Constant
+}
+
+// Create creates the Gorgonia weight initializer from this
+// initializer config
+func (c ConstantConfig) Create() G.InitWFn {
+	return G.ValuesOf(c.Value)
+}

initwfn/InitWFn.go

@@ -16,12 +16,13 @@ type Type string
 
 // Available InitWFn types
 const (
-	GlorotU Type = "GlorotU"
-	GlorotN Type = "GlorotN"
-	HeU     Type = "HeU"
-	HeN     Type = "HeN"
-	Zeroes  Type = "Zeroes"
-	Ones    Type = "Ones"
+	GlorotU  Type = "GlorotU"
+	GlorotN  Type = "GlorotN"
+	HeU      Type = "HeU"
+	HeN      Type = "HeN"
+	Zeroes   Type = "Zeroes"
+	Ones     Type = "Ones"
+	Constant Type = "Constant"
 )
 
 // InitWFn wraps Gorgonia InitWFn so that they can be JSON marshalled and
@@ -57,12 +58,13 @@ func (i *InitWFn) UnmarshalJSON(data []byte) error {
 		"Type",
 		"Config",
 		map[string]reflect.Type{
-			string(GlorotU): reflect.TypeOf(GlorotUConfig{}),
-			string(GlorotN): reflect.TypeOf(GlorotNConfig{}),
-			string(HeU):     reflect.TypeOf(HeUConfig{}),
-			string(HeN):     reflect.TypeOf(HeNConfig{}),
-			string(Zeroes):  reflect.TypeOf(ZeroesConfig{}),
-			string(Ones):    reflect.TypeOf(OnesConfig{}),
+			string(GlorotU):  reflect.TypeOf(GlorotUConfig{}),
+			string(GlorotN):  reflect.TypeOf(GlorotNConfig{}),
+			string(HeU):      reflect.TypeOf(HeUConfig{}),
+			string(HeN):      reflect.TypeOf(HeNConfig{}),
+			string(Zeroes):   reflect.TypeOf(ZeroesConfig{}),
+			string(Ones):     reflect.TypeOf(OnesConfig{}),
+			string(Constant): reflect.TypeOf(ConstantConfig{}),
 		})
 	if err != nil {
 		return err