Updated the Zeroes() weight ini t function (#422)

gorgonia · Jul 21, 2020 · 57e4244 · 57e4244
1 parent 5cd5dc0
commit 57e4244
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/weights.go b/weights.go
@@ -2,6 +2,7 @@ package gorgonia
 
 import (
 	"math"
+	"reflect"
 	"time"
 
 	rng "github.com/leesper/go_rng"
@@ -30,8 +31,7 @@ func Zeroes() InitWFn {
 		case tensor.Int:
 			return make([]int, size)
 		default:
-			err := errors.Errorf(nyiTypeFail, "Zeroes", dt)
-			panic(err)
+			return reflect.MakeSlice(reflect.SliceOf(dt.Type), size, size).Interface()
 		}
 	}
 	return f