tf.cast\tf.argmax\tf.argmin

1. tf.cast()

​​tf.cast()​​的作用是将一个张量的类型改变为另外一个类型，如第11行，将浮点型转化为整数型

def cast(x, dtype, name=None):
  """Casts a tensor to a new type.

  The operation casts `x` (in case of `Tensor`) or `x.values`
  (in case of `SparseTensor`) to `dtype`.

  For example:

  ```python
  x = tf.constant([1.8, 2.2], dtype=tf.float32)
  tf.cast(x, tf.int32)  # [1, 2], dtype=tf.int32
  ```

2. tf.argmax()

def argmax(input,
           axis=None,
           name=None,
           dimension=None,
           output_type=dtypes.int64):

​​tf.argmax()​​​的作用是选择某个维度中，最大值所对应的下标（索引）；当​​axis=1​​时，表示在每一行选取最大值对应的下标。

例如：

a=tf.constant([1,2,3,4,5,0],dtype=tf.int32)
tf.argmax(a,axis=1) # 4

3. tf.argmin()

理解同​​argmax()​​