The error is probably thrown from a matrix multiplication (the stacktrace will show you what exactly is causing it). For matrix multiplication, you need the dimensions to be (N, M) * (M, L) (i.e. the inner dimension must match). You have (N, M) * (N, M) which doesn’t work (inner dimensions don’t match). One of those matrices should be probably be transposed.
1 Like