Refactor Rust ops code

[wilhelmagren]

Redundant test code should be removed.
Documentation for functions should be added.
More ops need to be implemented (e.g, dot prod, matmul, relu, etc...)