Failing tests in `CrossEntropySoftmax` and `GramSchmidt` initializer

[AleksaMCode]

I’ve discovered three failing tests: one gramschmidt initializer case likely failing the orthonormality assertion, and two CrossEntropySoftmax cases very likely failing in the SMPO apply/transpose path due to a tensor shape-permutation mismatch. Error message:

FAILED test/test_initializers.py::test_gramschmidt_init[normal-0.01-shape4] - AssertionError: assert False
FAILED test/test_metrics.py::test_CrossEntropySoftmax - TypeError: transpose permutation isn't a permutation of operand dimensions, got permutation (0,) for operand shape (3, 1).
FAILED test/test_metrics.py::test_CrossEntropySoftmax_with_embedded_numpy_array - TypeError: transpose permutation isn't a permutation of operand dimensions, got permutation (0,) for operand shape (3, 1).

[emapuljak]

Thanks! Will update them soon.

[emapuljak]

Now tests are passing with new PR: #37 --> will be merged in next few days.