Tensorflow常見函數case argmax equal

經常使用的函數:ide

tf.argmax(input, axis=None, name=None, dimension=None)

  • input:輸入Tensor
  • axis:0表示按列,1表示按行
  • name:名稱
  • 返回的爲索引
    1 import tensorflow as tf
    2 import numpy as np
    3 
    4 A , B  = [[1, 3, 4, 5, 6]],[[1, 3, 4], [2, 4, 1]]
    5 with tf.Session() as sess:
    6     print(sess.run(tf.argmax(A, 1)))
    7     print(sess.run(tf.argmax(B, 1)))
    tf.argmax

    結果:函數

  • [4]
    [2 1]spa

tf.case(pred_fn_pairs, default=None, exclusive=False, strict=False, name='case')

  • pred_fn_pairs參數是大小爲N的字典或pairs的列表--      由boolean標量和可調用函數返回張量列表
  • exclusive==True,則計算全部的謂詞,若是多個謂詞計算爲True,則引起異常。
  • default: 默認返回tensors列表
  • strict: boolean打開或關閉‘strict‘ 模式

 

tf.cast(data,tf.float32)

  • 數據類型轉化(data轉化爲float32類型)

tf.equal(A, B)是對比這兩個矩陣或者向量的相等的元素,若是是相等的那就返回True,反正返回False,返回的值的矩陣維度和A是同樣的

代碼:code

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

結果:[[ True  True  True False False]]blog

相關文章
相關標籤/搜索