阅读(5.4k) 书签 (0)

TensorFlow函数教程:tf.keras.backend.argmin

2019-03-06 16:10 更新

tf.keras.backend.argmin函数

tf.keras.backend.argmin(
    x,
    axis=-1
)

定义在:tensorflow/python/keras/backend.py。

返回沿轴的最小值索引。

参数:

  • x:张量或变量。
  • axis:沿其执行缩小的轴。

返回:

该函数返回一个张量。