Comment by GaggiX

Hopefully the library will support in the future the shapes of the tensors in the type system, that would be great to prevent many errors.