文件名称：tf_custom_op:用于向TensorFlow添加自定义操作的样板模板
文件大小：11KB
文件格式：ZIP
更新时间：2024-05-21 08:09:16
deep-learning cpp tensorflow C++
TensorFlow的自定义操作 这将很快被弃用。 无需从源代码编译TensorFlow。 这是在TensorFlow中添加自定义C ++ / CUDA操作的非常简单的示例，其预期用途只是其他自定义TensorFlow操作的起点。 当前版本已在TensorFlow v1.9上进行了测试。 运行脚本 cd user_ops cmake . make python test_matrix_add.py
【文件预览】：
tf_custom_op-master
----example.py(534B)
----setup.cfg(74B)
----.gitignore(1KB)
----README.md(540B)
----user_ops()
--------FindTENSORFLOW.cmake(4KB)
--------test_matrix_add.py(2KB)
--------__init__.py(671B)
--------ops()
--------.gitignore(21B)
--------CMakeLists.txt(2KB)
--------kernels()