support more than one batch dimension #2
Description
could be as simple as reshaping the input arrays before calling the kernel. NNlib would need similar changes to be consistent.
{{ message }}
could be as simple as reshaping the input arrays before calling the kernel. NNlib would need similar changes to be consistent.