Tensflow的equal函数

一、函数 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 
posted @ 2018-01-04 23:05  AI菌  阅读(606)  评论(0编辑  收藏  举报