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.PyTorch C/CUDA Template on GitHub