Tensorflow
LauJames
这个作者很懒,什么都没留下…
展开
-
利用tf.ragged.boolean_mask完成选择mask压缩tensor
现有如下输入:1. [Batch_size, Seq_len, N_classes]的tensor T2. [Batch_size, Seq_len]的mask矩阵 M需求:根据M中的值来mask T,并去掉被mask掉的值实际场景:主要是现在有一个对话的每个时刻的状态,和角色(0/1),需要取出角色为1的所有utterances的状态,在此基础上选取最后一个角色为1的utterance的状态问题点:若使用tf.boolean_mask,则会得到[?, N_classes], ?代表原创 2021-04-30 17:35:34 · 387 阅读 · 1 评论 -
tf.contrib.layers.dropout和tf.nn.dropout的不同导致的bug
今天在写RNN的时候,Dense layer需要添加一个dropout,至于keep_prob按道理是需要用一个placeholder:self.dropout_keep_prob = tf.placeholder(tf.float32, name='keep_prob')来占位,以便于在训练/测试的时候区分。利用feed_dict 来传递 keep_prob的值 看别人的教程,一模一样原创 2018-01-21 18:00:12 · 4588 阅读 · 9 评论 -
tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度
tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度在利用tensorflow进行文本挖掘工作的时候,经常涉及到维度扩展和压缩工作。比如对文本进行embedding操作完成之后,若要进行卷积操作,就需要对embedded的向量扩展维度,将[batch_size, embedding_dims]扩展成为[batch_size, embedding_dims,...原创 2018-05-14 15:41:58 · 9021 阅读 · 3 评论 -
Tensorflow版本和cudnn版本不对应问题解决
E tensorflow/stream_executor/cuda/cuda_dnn.cc:343] Loaded runtime CuDNN library: 7.0.5 but source was compiled with: 7.2.1. CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version.原创 2019-01-11 11:55:04 · 6445 阅读 · 2 评论 -
tf.where 和 tf.cond对比
tf.where 和 tf.cond都是类似于if…else 的逻辑操作,现对比两者区别import tensorflow as tf pred=tf.placeholder(dtype=tf.bool,name='bool') x=tf.constant(1) y = tf.cond(pred,lambda:x+1,lambda:x-1) z=tf.where(pred,x+1,x-...转载 2019-01-17 16:54:56 · 2352 阅读 · 0 评论