joelberkeley/spidr

Allow `reduce` across multiple axes at once

Closed this issue · 0 comments

joelberkeley commented

This is mostly a types problem. XLA implementation is trivial. For example, trace would simplify to

reduce @{Sum} [0, 1] (x * identity)