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

tf.keras.backend.argmax函数

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

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

返回沿轴的最大值索引。

参数:

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

返回:

该函数返回一个张量。

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部