cupy.argmin

cupy.argmin(a, axis=None, dtype=None, out=None, keepdims=False)

返回沿轴的最小值的索引。

参数:

• a (cupy.ndarray) – 要获取最小值的数组。

• axis (int) – 沿哪个轴查找最小值。 默认情况下，a 会被展平。

• dtype – 数据类型指定符。

• out (cupy.ndarray) – 输出数组。

• keepdims (bool) – 如果为 True，则轴 axis 将被保留为一个长度为一的轴。

返回:

沿轴的 a 的最小值的索引。

返回类型:

cupy.ndarray

注意

dtype 和 keepdim 参数是 CuPy 特有的。它们在 NumPy 中不存在。

注意

axis 参数接受一个整数元组，但这是 CuPy 特有的。NumPy 不支持此特性。

另请参阅

numpy.argmin()