Travis CIGithub CI

] add ArrayAllez

log! ∘ exp!

This began as a way to more conveniently choose between Yeppp! and AppleAccelerate and IntelVectorMath, without requiring that any by installed. The fallback version is just a loop, with @threads for large enough arrays.

x = rand(1,100);

y = exp0(x)  # precisely = exp.(x)
x  log!(y)  # in-place, just a loop

using AppleAccelerate  # or using IntelVectorMath, or using Yeppp

y = exp!(x)  # with ! mutates
x = log_(y)  # with _ copies

Besides log! and exp!, there is also scale! which understands rows/columns. And iscale! which divides, and inv! which is an element-wise inverse. All have non-mutating versions ending _ instead of !, and simple broadcast-ed versions with 0.

m = ones(3,7)
v = rand(3)
r = rand(7)'

scale0(m, 99)  # simply m .* 99
scale_(m, v)   # like m .* v but using rmul!
iscale!(m, r)  # like m ./ r but mutating.

These commands all make some attempt to define gradients for use with Tracker ans Zygote, but caveat emptor. There is also an exp!! which mutates both its forward input and its backward gradient, which may be a terrible idea.

using Tracker
x = param(randn(5));
y = exp_(x)

Tracker.back!(sum_(exp!(x))) == y # true

This package also defines gradients for prod (overwriting an incorrect one) and cumprod, as in this PR.


An experiment with LRUCache for working space:

x = rand(2000)' # turns off below this size

copy_(:copy, x)
similar_(:sim, x)
Array_{Float64}(:new, 5,1000) # @btime 200 ns, 32 bytes

inv_(:inv, x) # most of the _ functions can opt-in


This macro wraps reductions like sum(A; dims=...) in dropdims(). It understands things like this:

@dropdims sum(10 .* randn(2,10); dims=2) do x
    trunc(Int, x)


This package used to provide two functions generalising matrix multiplication. They are now better handled by other packages:

  • TensorCore.boxdot contracts neighbours: rand(2,3,5) ⊡ rand(5,7,11) |> size == (2,3,7,11)
  • NNlib.batched_mul keeps a batch dimension: rand(2,3,10) ⊠ rand(3,5,10) |> size == (2,5,10)

See Also