如何在tensorflow中判断tensor(张量)的值
众所周知,在tensorflow程序中,tensor只是占位符,在会话层没有run出tensor的值之前,我们是无法获知tensor的值的(不过笔者听说tensorflow要改良这一机制,将tensor转化为可以判断值的数据结构-flag:2017.11.13)。也就是说,下面的逻辑是无法实现的:
#a is a 1-D tensor, b is a 1-D tensor too. if a == b: #take some actions... else: #take some other actions...
可是,我们在程序中有可能需要根据tensor的值作出判断,以采取不同的操作。笔者就曾经遇到过这种问题,笔者在构造一个网络时,网络的输入为两个tensor,在这里分别称为a和b,需要根据b的值对a采取不同的操作。可是,在程序中是无法通过逻辑语句去判断tensor b的值的。那么,我们应该怎么办呢?
对于这个问题,tensorflow官方已经为我们提供了一个能满足功能的接口,即tf.cond,如下图所示。
相关推荐
xiexf 2019-12-20
songbinxu 2019-12-08
zhuzhishi0 2019-09-08
AI技术研究院 2019-07-01
hackerlong 2019-06-21
aklk 2017-03-02
白话深度学习 2018-07-08
daixiang 2018-07-30
源式羽语 2020-01-24
MasterCui 2019-09-17
qiujiahao 2019-07-01
ruyulin 2018-10-28
Donie 2017-06-20
sherry颖 2019-01-03
深度学习图像算法 2018-10-06
dxmkkk 2018-05-23
Kindle君 2018-09-13
lingyu 2017-01-18
zxyscz 2018-02-15