Add Must to rand package

layer/linear.go

@@ -17,7 +17,7 @@ type LinearOpts struct {
 }
 
 func Linear(outSize int, opts ...LinearOpts) *LinearT {
-	s := rand.NewSource()
+	s := rand.MustNewSource()
 	if len(opts) != 0 && opts[0].Source != nil {
 		s = opts[0].Source
 	}

matrix/matrix.go

@@ -46,7 +46,7 @@ func From(x [][]int) Matrix {
 // rnd returns a pseudo-random number generator.
 func rnd(s ...randv2.Source) *randv2.Rand {
 	if len(s) == 0 {
-		s = append(s, rand.NewSource())
+		s = append(s, rand.MustNewSource())
 	}
 
 	return randv2.New(s[0])

rand/source.go

@@ -2,17 +2,30 @@ package rand
 
 import (
 	crand "crypto/rand"
+	"fmt"
 	randv2 "math/rand/v2"
 )
 
+func Must[T any](a T, err error) T {
+	if err != nil {
+		panic(err)
+	}
+
+	return a
+}
+
+func MustNewSource() randv2.Source {
+	return Must(NewSource())
+}
+
 // NewSource returns a source of pseudo-random number generator
-func NewSource() randv2.Source {
+func NewSource() (randv2.Source, error) {
 	var p [32]byte
 	if _, err := crand.Read(p[:]); err != nil {
-		panic(err)
+		return nil, fmt.Errorf("read: %v", err)
 	}
 
-	return randv2.NewChaCha8(p)
+	return randv2.NewChaCha8(p), nil
 }
 
 // Const returns a source of constant pseudo-random number generator

rand/source_test.go

@@ -35,8 +35,8 @@ func ExampleConst() {
 	// 0.6764556596678
 }
 
-func TestNewSource(t *testing.T) {
-	v := randv2.New(rand.NewSource()).Float64()
+func TestMustNewSource(t *testing.T) {
+	v := randv2.New(rand.MustNewSource()).Float64()
 	if v >= 0 && v < 1 {
 		return
 	}