OPEN SOURCE

Custom Tensorflow Operations in C++

This project includes an example of implementing a custom TensorFlow operation in C++. The repository implements a simple linear operation, the innter product with some parameters. As such, the operation includes trainable parameters as well as gradients.

Also check out this article.