Merge pull request #13 from rmsrosa/dev

support math functions on NeuralNumbers

.github/workflows/ci.yml

@@ -5,7 +5,7 @@ on:
       - main
   pull_request:
     branches:
-      - master
+      - main
 jobs:
   test:
     name: Julia ${{ matrix.version }} - ${{ matrix.os }} - ${{ matrix.arch }} - ${{ github.event_name }}

Project.toml

@@ -1,7 +1,7 @@
 name = "ChainPlots"
 uuid = "43761bbe-b505-42c3-b8bd-5d4ae40fd227"
 authors = ["Ricardo Rosa <[email protected]>"]
-version = "0.1.2"
+version = "0.1.3"
 
 [deps]
 Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"

src/NeuralNumbers.jl

@@ -98,17 +98,22 @@ Base.promote_rule(::Type{NeuralNumber}, ::Type{<:Number}) = NeuralNumber
 
 Random.rand(::Random.AbstractRNG, ::Random.Sampler{NeuralNumber}) = cold
 
-for f in [:+, :-, :abs, :abs2, :inv, :tanh, :sqrt,
-    :exp, :log, :log1p, :log2, :log10,
-    :conj, :transpose, :adjoint, :angle]
+for f in (:log, :exp, :sin, :cos, :tan, :sinh, :cosh, :tanh, :asin,
+    :acos, :atan, :asinh, :acosh, :atanh, :cbrt, :sqrt, :log2, :log10,
+    :max, :min, :exp2, :exp10, :expm1, :log1p, :sinpi, :cospi, :mod2pi,
+    )
     @eval Base.$f(x::NeuralNumber) = x
 end
 
-for f in [:+, :-, :*, :/, :^, :mod, :div, :rem, :widemul]
+for f in (:+, :-, :abs, :abs2, :inv, :conj, :transpose, :adjoint, :angle)
+    @eval Base.$f(x::NeuralNumber) = x
+end
+
+for f in (:+, :-, :*, :/, :^, :mod, :div, :rem, :widemul)
     @eval Base.$f(x::NeuralNumber, y::NeuralNumber) = max(x, y)
 end
 
-for f in [:+, :-, :*, :/, :^, :mod, :div, :rem, :widemul]
+for f in (:+, :-, :*, :/, :^, :mod, :div, :rem, :widemul)
     # specialize to avoid conflict with Base
 #=     @eval Base.$f(x::NeuralNumber, ::Bool) = x
     @eval Base.$f(::Bool, y::NeuralNumber) = y =#