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()

猜你喜欢

转载自blog.csdn.net/the_lastest/article/details/81050778