Update some files

itsubaki · Oct 21, 2023 · 014b7e6 · 014b7e6
1 parent b949a9d
commit 014b7e6
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/optimizer/sgd.go b/optimizer/sgd.go
@@ -1,8 +1,6 @@
 package optimizer
 
-import (
-	"github.com/itsubaki/autograd/matrix"
-)
+import "github.com/itsubaki/autograd/matrix"
 
 // SGD is an optimizer that the Stochastic Gradient Descent algorithm.
 type SGD struct {

diff --git a/variable/variable.go b/variable/variable.go
@@ -24,6 +24,10 @@ func NewOf(v ...[]float64) *Variable {
 	return &Variable{Data: v}
 }
 
+func ZeroLike(v *Variable) *Variable {
+	return &Variable{Data: matrix.ZeroLike(v.Data)}
+}
+
 func OneLike(v *Variable) *Variable {
 	return &Variable{Data: matrix.OneLike(v.Data)}
 }

diff --git a/variable/variable_test.go b/variable/variable_test.go
@@ -42,6 +42,14 @@ func ExampleConst() {
 	// const([1])
 }
 
+func ExampleZeroLike() {
+	v := variable.New(1, 2, 3, 4)
+	fmt.Println(variable.ZeroLike(v))
+
+	// Output:
+	// variable([0 0 0 0])
+}
+
 func ExampleOneLike() {
 	v := variable.New(1, 2, 3, 4)
 	fmt.Println(variable.OneLike(v))