def argmin(x, axis=-1):
"""Returns the index of the minimum value along an axis.
# Arguments
x: Tensor or variable.
axis: axis along which to perform the reduction.
# Returns
A tensor.
"""
axis = _normalize_axis(axis, ndim(x))
return tf.argmin(x, axis)
tensorflow_backend.py 文件源码
python
阅读 28
收藏 0
点赞 0
评论 0
评论列表
文章目录