From a7e175db7a53e91026d4bfea3542045494a54b24 Mon Sep 17 00:00:00 2001
From: itsubaki <1759459+itsubaki@users.noreply.github.com>
Date: Sat, 14 Oct 2023 10:46:54 +0900
Subject: [PATCH] Update Matrix

---
 matrix/matrix.go | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/matrix/matrix.go b/matrix/matrix.go
index d2cd817..a1bf8f6 100644
--- a/matrix/matrix.go
+++ b/matrix/matrix.go
@@ -210,14 +210,12 @@ func BroadcastTo(shape []int, m Matrix) Matrix {
 	a, b := shape[0], shape[1]
 
 	if len(m) == 1 && len(m[0]) == 1 {
-		out := Zero(a, b)
-		for i := 0; i < a; i++ {
-			for j := 0; j < b; j++ {
-				out[i][j] = m[0][0]
-			}
+		out := make([]float64, a*b)
+		for i := 0; i < a*b; i++ {
+			out[i] = m[0][0]
 		}
 
-		return out
+		return Reshape(shape, New(out))
 	}
 
 	if len(m) == 1 {
@@ -340,10 +338,9 @@ func Reshape(shape []int, m Matrix) Matrix {
 		b = p * q / a
 	}
 
-	out := New()
+	out := make(Matrix, a)
 	for i := 0; i < a; i++ {
-		begin, end := i*b, (i+1)*b
-		out = append(out, v[begin:end])
+		out[i] = v[i*b : (i+1)*b]
 	}
 
 	return out