Я хотел бы использовать в модели Pytorch (или Tensorflow, если он работает лучше) собственный слой C ++ / CUDA Caffer.Я не знаю, есть ли способ сделать это, или я должен переопределить весь слой в Pytorch / Tensorflow.Это репозиторий с пользовательской реализацией caffe: https://github.com/MPI-IS/bilateralNN.
Спасибо!