
Tensorflow
raintungli
这个作者很懒,什么都没留下…
展开
-
Tensorflow 源码分析-会话与线程池之间的关系
1. Tensorflow 的sessionFactory创建新的会话,tensorflow使用了多工厂模式,在不同的场景下使用不同的工厂, 使用什么工厂模式由传递进来的SessionOptions来决定。1.1 注册工厂Tensorflow 提供了可以注册多会话工厂的模式,允许不同的模块注册自己的会话工厂void SessionFactory::Register(const string&...原创 2018-04-02 17:11:16 · 4290 阅读 · 0 评论 -
Tensorflow 源码分析-GPU调用是如何实现的
1. Tensorflow GPU支持Tensorflow 支持GPU进行运算,目前官方版本只支持NVIDIA的GPU,可以在tensorflow的官方上看到。Tensorflow 对GPU的运算的支持最小力度就是OP,也就是我们常说的算子,下图提供了Tensorflow的一些常见算子,而每个算子在Tensorflow上都会提供GPU的算法:关于OP的具体实现,在本篇博客中就不叙述了。2. Ten...原创 2017-12-21 10:22:30 · 9727 阅读 · 1 评论 -
【TensorFlow基础函数】tf.transpose函数说明和用法
tf.transpose 函数官网介绍https://www.tensorflow.org/api_docs/python/tf/transposetf.transpose( a, perm=None, name='transpose', conjugate=False)函数说明:依据指定perm交换输入张量的不同维度参数perm: ...原创 2018-01-23 14:58:12 · 1343 阅读 · 0 评论 -
Tensorflow 源码分析- 从GPU OOM开始说Tensorflow的BFC内存管理
前言在平台上跑GPU训练,结果CUDA OOM了,错误提示E Internal: failed initializing StreamExecutor for CUDA device ordinal 0: Internal: failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_OUT_OF_MEMORY; total memory repor...原创 2018-05-04 11:46:13 · 8572 阅读 · 1 评论 -
【TensorFlow基础函数】tf.concat的用法
tf.concat 的用法TF官方的文档tf.concat( values, axis, name='concat')连接多个Tensor的操作values 多个Tensoraxis是哪个纬度0 代表第一纬度1 代表第二纬度2 代表第三纬度n 代表第n+1纬度案例: t1 = [[[1, 2, 3], [4, 5, 6]],[[7, 8, 9...原创 2019-02-25 21:44:31 · 2369 阅读 · 0 评论 -
用Bert模型计算句子的vector
1. Bert模型Bert模型是2018年google开源出来的NLP的模型,具体实现请参考:https://github.com/google-research/bert2. Bert模型计算句子的vector2.1 通过模型获取句子的encoder层 model = modeling.BertModel( config=bert_config, ...原创 2019-07-21 17:57:24 · 3203 阅读 · 0 评论