Tensflow的equal函数

Posted 2020-10-19 AI菌

一、函数 tf.equal()

equal(
    x,
    y,
    name=None
)

tf.equal(A, B)是对比这两个矩阵或者向量的相等的元素，如果是相等的那就返回True，反正返回False，返回的值的矩阵维度和A是一样的

 

二、用法

【code】

import tensorflow as tf  
import numpy as np  
  
A = [[1,3,4,5,6]]  
B = [[1,3,4,3,2]]  
  
with tf.Session() as sess:  
    print(sess.run(tf.equal(A, B))) 

【result】

[[ True  True  True False False]]

 

------------------------------------------------------

转自：

1. http://blog.csdn.net/uestc_c2_403/article/details/72232924