Tensorflow: Schreiben einer Op in Python
Ich würde gerne eine Op in Python. Dieses tutorial nur erklärt wie man es in c++ mit Python-wrapper.
https://www.tensorflow.org/versions/master/how_tos/adding_an_op/index.html#adding-a-new-op
Wie kann ich schreiben das es komplett in Python?
Du musst angemeldet sein, um einen Kommentar abzugeben.
Können Sie
tf.py_func(func inp, Tout)
.Ihre python-Funktion zu haben:
inp
Tout
Innerhalb der Funktion können Sie tun, was Sie wollen, wenn Bedingungen für for-Schleifen, was ist nicht möglich in TensorFlow.
Jedoch, der Vorgang wird ausgeführt, auf die CPU-es kann also langsamer sein als das äquivalent TensorFlow op in GPU.
Können Sie
tf.py_func
zu nennen python-Funktionen. Die Operationen innerhalb der Funktion können Sie auch auf der GPU. Zum Beispiel, wir können hinzufügen, eine Op und deren Verlauf rein in python, welche Anrufe Caffe auf GPU: