For this I needed a good notation for functions applied to specific dimensions and broadcasting over the rest. Like softmax in a transformer.
The function chapter is still under development in the book though. So if you have any good references for how it's been done graphically in the past, that I might have missed, feel free to share them.
Tensors, as used in ML, are much closer to a key-value store with composite keys and scalar values, with most of the complexity coming from deciding how to filter on those composite keys.
Drop me a line if you're interested in a chat. This is something I've been thinking about for years now.
https://arxiv.org/abs/2302.09687
(On functions of 3rd-order "tensors")
((Whereas matrix-functions are of 2nd-order "tensors"))
Playground: https://gitlab.com/katlund/t-frechet
(MATLAB)
Some confusion arises from the difference between f:R -> R and f':R -> R. It's Fréchet derivative is Df:R -> L(R,R) where Df(x)h = f'(x)h. Row vectors and column vectors a just a clumsy way of thinking about this.
BTW, all you need in order to publish on arixv.org is to know a FoF. There is no rigorous peer review. https://arxiv.org/abs/1912.01091, https://arxiv.org/abs/2009.10852.