You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.ops.Argmin.rst 837 B

1234567891011121314151617181920212223242526
  1. mindspore.ops.Argmin
  2. =====================
  3. .. py:class:: mindspore.ops.Argmin(axis=-1, output_type=mstype.int32)
  4. 返回输入Tensor在指定轴上的最小值索引。
  5. 如果输入Tensor的shape为 :math:`(x_1, ..., x_N)` ,则输出Tensor的shape为 :math:`(x_1, ..., x_{axis-1}, x_{axis+1}, ..., x_N)` 。
  6. **参数:**
  7. - **axis** (int) - 指定Argmin计算轴。默认值:-1。
  8. - **output_type** (:class:`mindspore.dtype`) - 指定输出数据类型。默认值: `mindspore.dtype.int32` 。
  9. **输入:**
  10. - **input_x** (Tensor) - Argmin的输入,任意维度的Tensor。
  11. **输出:**
  12. Tensor,输出为指定轴上输入Tensor最小值的索引。
  13. **异常:**
  14. - **TypeError** - `axis` 不是int。
  15. - **TypeError** - `output_type` 既不是int32也不是int64。