def argmin(x, axis=-1):
"""Returns the index of the minimum value along an axis.
# Arguments
x: input tensor.
axis: axis along which to perform the reduction.
keepdims: whether the drop or broadcast the reduction axes.
# Returns
A tensor.
"""
if axis < 0:
axis = axis % len(x.get_shape())
return tf.argmin(x, axis)
tensorflow_backend.py 文件源码
python
阅读 31
收藏 0
点赞 0
评论 0
评论列表
文章目录