Open Ericgig opened 3 months ago
Hi @Ericgig ! Have you been able to find the reason for this particular issue?
No.
I guess is JaxArray(mul_jax(self.data, other))
is seen as a 1x1 matrix when tracer are used instead of real array. But I don't know why.
Input
Output: