Tensorflow-Go的扩展

最新推荐文章于 2024-06-19 09:33:21 发布

InsZVA

最新推荐文章于 2024-06-19 09:33:21 发布

阅读量1.5k

点赞数 1

分类专栏：算法文章标签： tensorflow go 深度学习

本文链接：https://blog.youkuaiyun.com/InsZVA/article/details/78744001

版权

本文揭示了尽管Tensorflow官方Go库主要用于加载Python创建的模型，但其实可以通过实验发现可以创建和使用Variable操作。通过封装梯度计算的OP并修改变量，可以在Go中实现类似Python版本的完整功能。同时，由于官方库缺少contrib模块，文章还提到了如何在编译时添加相应依赖来解决这一问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

谷歌的tensorflow虽然提供了go版本，但是官方的说法是：

TensorFlow provides APIs for use in Go programs. These APIs are particularly well-suited to loading models created in Python and executing them within a Go application.

意思是go的库只是用来装载python创建的模型，然后执行的，而且在go版本api的godoc中也写到：

The tensorflow package currently does not have the ability to export a model to a directory from Go. This function thus currently targets loading models exported in other languages, such as using tf.saved_model.builder in Python. See: https://www.tensorflow.org/code/tensorflow/python/saved_model/

说go不能将模型导出，而且现阶段go版本的api没有直接创建variable的op，但是通过实验可以发现其实是可以使用的，先看/tensorflow/core/ops/state_ops.cc中variable这个op的声明：