tensorflow中判定两tensor是否完全相等

故事已经翻了几页

已于 2023-02-10 18:16:10 修改

阅读量1.8k

点赞数

分类专栏： tensorflow探索文章标签： tensorflow python 人工智能

于 2020-12-09 16:36:39 首次发布

本文链接：https://blog.csdn.net/gushiyi27/article/details/110929443

版权

本文介绍了在TensorFlow中如何判断两个Tensor是否完全相等，重点讲解了如何构造条件函数来实现这一操作，避免了直接使用if语句导致的错误。文章通过对比numpy中的(narray1 == narray2).all()方法，逐步解析了在TensorFlow中实现相同功能的cond3()函数的工作原理。

摘要由CSDN通过智能技术生成

tensorflow中判定两tensor是否完全相等

作用：等价于(narray1 == narray2).all()

1.tensorflow中不能使用while/if/for等，ta的循环语句为tf.while_loop()(笔者这里走了弯路，在没学习tensorflow基本语法时就开始了构建模型T_T，结果运行时报错，找了好久才找到错误，望读者吸取教训╮(￣▽ ￣)╭)。
2.要使用tf.while_loop函数取代while…break,便需要先构造cond()、body()这两个函数，然后使用x = tf.while_loop(cond, body, x)。while_loop()的详细使用方法这里不再赘述，网上已经有很多教程了。

在numpy中判定两个矩阵是否完全相等的代码是(narray1 == narray2).all()，如果narray1和narray2在每个对应的位置上都相等则返回true，反之返回false。这段代码常用的格式如下：

#代码1
while True:
    ....
    if (narray1 == narray2).all():
        break
    ...

在tensorflow中要写成

#代码2
def cond():
    return a #f返回只有一个值的tensorflow的bool类型张量
def body():
    return b
x = tf.while_loop(cond, body, x)

其中要用后者cond()代替前者if。
下面介绍cond的构造方法。

最低0.47元/天解锁文章