Skip to main content
Max PlatformDocsCompanyBlogModCon
Max PlatformDocsCompanyBlogModCon
Contact salesSign in

function

argmax

argmax(input: NDBuffer[type, rank, shape, address_space], axis: Int, output: NDBuffer[type, rank, shape, address_space])

Finds the indices of the maximum element along the specified axis.

Args:

  • ​input (NDBuffer[type, rank, shape, address_space]): The input tensor.
  • ​axis (Int): The axis.
  • ​output (NDBuffer[type, rank, shape, address_space]): The output tensor.

argmax(input: NDBuffer[type, rank, shape, address_space], axis_buf: NDBuffer[type, rank, shape, address_space], output: NDBuffer[type, rank, shape, address_space])

Finds the indices of the maximum element along the specified axis.

Args:

  • ​input (NDBuffer[type, rank, shape, address_space]): The input tensor.
  • ​axis_buf (NDBuffer[type, rank, shape, address_space]): The axis tensor.
  • ​output (NDBuffer[type, rank, shape, address_space]): The axis tensor.