Mit einfachen mathematischen Funktionen bei der Implementierung neuer Op in Tensorflow

Question

Ich versuche, eine neue Op in C++ durch Schreiben einer .cc-Datei (CPU-und GPU-Versionen) zu implementieren. Dieses Op verwendet einfache mathematische Funktionen wie cos oder sin.

Was ist der richtige Weg, cos oder sin (oder dergleichen) auf Floats in CPU/GPU aufzurufen. Müssen wir eine externe Bibliothek hinzufügen? Ist es in Eigen? Müssen wir verschiedene Bibliotheken für CPU und GPU verwenden? Wie deklariert man es?

Gibt es weitere Dokumentationen/Beispiele, die Sie kennen außer zero_out.cc?

Answer 1

Ich verwendete std::cos und std::sin und es funktioniert gut ON CPU! Keine Ahnung, ob es so sein sollte, aber ich werde diese Antwort behalten, bis jemand es widerlegt.

EDIT: Nun, es scheint, ich habe zu schnell gesprochen: die GPU-Version benötigt CUDA-Funktion und nicht C++ - Funktionen (so genannte Host-Funktionen).