Fix softmax op (#506)

* Fix go mod tidy command * Fix SoftMax operation Now it uses the tensor implementation
gorgonia · Oct 21, 2021 · 70448c3 · 70448c3
1 parent 68fd765
commit 70448c3
Show file tree

Hide file tree

Showing 26 changed files with 186 additions and 2,718 deletions.
diff --git a/example_err_test.go b/example_err_test.go
@@ -61,9 +61,9 @@ func Example_errorHandling() {
 	_ = nn2PlusWrong
 
 	// Output:
-	// nn: ÷ false(%a, %f) :: Matrix float32
+	// nn: Softmax{-1, false}()(%9) :: Matrix float32
 	// An error occurs: Type inference error. Op: + false. Children: [Matrix float32, Matrix float64], OpType:Matrix a → Matrix a → Matrix a: Unable to unify while inferring type of + false: Unification Fail: float64 ~ float32 cannot be unified
-	// nn2: ÷ false(%a, %f) :: Matrix float32
+	// nn2: Softmax{-1, false}()(%9) :: Matrix float32
 	// An error occurs (caught by recover()): Type inference error. Op: + false. Children: [Matrix float32, Matrix float64], OpType:Matrix a → Matrix a → Matrix a: Unable to unify while inferring type of + false: Unification Fail: float64 ~ float32 cannot be unified
 
 }
diff --git a/examples/charRNN/go.mod b/examples/charRNN/go.mod
diff --git a/examples/charRNN/go.sum b/examples/charRNN/go.sum
diff --git a/examples/convnet/go.mod b/examples/convnet/go.mod
diff --git a/examples/convnet/go.sum b/examples/convnet/go.sum
diff --git a/examples/convnet_cuda/go.mod b/examples/convnet_cuda/go.mod
diff --git a/examples/convnet_cuda/go.sum b/examples/convnet_cuda/go.sum
diff --git a/examples/iris/go.mod b/examples/iris/go.mod
diff --git a/examples/iris/go.sum b/examples/iris/go.sum
diff --git a/examples/logisticregression/go.mod b/examples/logisticregression/go.mod
diff --git a/examples/logisticregression/go.sum b/examples/logisticregression/go.sum
diff --git a/examples/mnist/go.mod b/examples/mnist/go.mod
diff --git a/examples/mnist/go.sum b/examples/mnist/go.sum
diff --git a/examples/resnet/go.mod b/examples/resnet/go.mod