ArgMin

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
Atlas A2系列产品	√
Atlas 200I/500 A2推理产品	×
Atlas推理系列产品	√
Atlas训练系列产品	√

功能说明

返回tensor中指定轴的最小值索引，并保存到out中。如果keepdim为false，则不保留对应的轴；如果为true，则保留指定轴的维度值为1。

函数原型

每个算子分为两段式接口，必须先调用“aclnnArgMinGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnArgMin”接口执行计算。

aclnnStatus aclnnArgMinGetWorkspaceSize(
  const aclTensor* self,
  int64_t          dim,
  bool             keepdim,
  aclTensor*       out,
  uint64_t*        workspaceSize,
  aclOpExecutor**  executor)

aclnnStatus aclnnArgMin(
  void          *workspace,
  uint64_t       workspaceSize,
  aclOpExecutor *executor,
  aclrtStream    stream)

aclnnArgMinGetWorkspaceSize

• 参数说明：

  参数名	输入/输出	描述	使用说明	数据类型	数据格式	维度(shape)	非连续Tensor
  self	输入	输入Tensor。	-	FLOAT、FLOAT16、BFLOAT16、DOUBLE、INT8、INT16、INT32、INT64、UINT8	ND	-	√
  dim	输入	需要进行最小值计算的维度。	取值范围在[-self.dim(), self.dim())。	-	-	-	-
  keepdim	输入	输出张量shape的维度是否与输入tensor的保持一致。	-	BOOL	-	-	-
  out	输出	输出Tensor。	-	INT32、INT64	ND	-	√
  workspaceSize	输出	返回需要在Device侧申请的workspace大小。	-	-	-	-	-
  executor	输出	返回op执行器，包含了算子计算流程。	-	-	-	-	-

  • Atlas训练系列产品：不支持BFLOAT16数据类型。
  • Atlas推理系列产品：不支持BFLOAT16数据类型。当self输入数据类型是INT8、INT16、INT32、INT64、UINT8时，输入张量shape的维度最大支持7维。

• 返回值：

  aclnnStatus：返回状态码，具体参见aclnn返回码。

  第一段接口完成入参校验，出现以下场景时报错：

  返回码	错误码	描述
  ACLNN_ERR_PARAM_NULLPTR	161001	传入的self或out是空指针。
  ACLNN_ERR_PARAM_INVALID	161002	self、out的数据类型和数据格式不在支持的范围之内。
  		推导出的数据类型无法转换为指定输出out的类型。
  		dim超出tensor数组维度。
  		传入的out的shape不满足infershape。

aclnnArgMin

• 参数说明：

  参数名	输入/输出	描述
  workspace	输入	在Device侧申请的workspace内存地址。
  workspaceSize	输入	在Device侧申请的workspace大小，由第一段接口aclnnArgMinGetWorkspaceSize获取。
  executor	输入	op执行器，包含了算子计算流程。
  stream	输入	指定执行任务的Stream。

• 返回值：

  aclnnStatus：返回状态码，具体参见aclnn返回码。

约束说明

• 确定性计算：

  • aclnnArgMin默认确定性实现。

• 输入self，不支持reduce轴为0的场景

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_arg_min	通过aclnnArgMin接口方式调用ArgMin算子。