axon icon indicating copy to clipboard operation
axon copied to clipboard

Published 20 hours ago • verified publisher icon elixir-nx

Remove ignore_batch logic in shape layers

Open seanmor5 opened this issue 2 years ago • 1 comments

It is just confusing, and I don't like the implicit behavior. It is easy enough for Axon to correctly do the reshape logic even with nil dimensions

seanmor5 avatar Jun 19 '22 14:06 seanmor5

I am considering adding batched_* variants of shape functions as an alternative to the option.

seanmor5 avatar Aug 08 '22 23:08 seanmor5