ArgMaxV13

trait ArgMaxV13 extends Operator
trait Operator
class Object
trait Matchable
class Any

Value members

Concrete methods

def ArgMaxV13[@specialized T <: UByte | UShort | UInt | ULong | Byte | Short | Int | Long | BFloat16 | Float16 | Float | Double : Numeric, Tt <: TensorTypeDenotation, Td <: TensorShapeDenotation, S <: Shape, Tt1 <: TensorTypeDenotation, Axis <: Index ::: INil, KeepDims <: Boolean & Singleton](name: String, axis: Axis, keepdims: KeepDims, selectLastIndex: Int, data: Tensor[T, (Tt, Td, S)])(implicit evidence$63: Numeric[T], tt: ValueOf[Tt1], td: TensorShapeDenotationOf[KeepOrReduceDimDenotations[Td, Axis, KeepDims]], s: ShapeOf[KeepOrReduceDims[S, Axis, KeepDims]], i: IndicesOf[Axis], k: ValueOf[KeepDims]): Tensor[Long, (Tt1, KeepOrReduceDimDenotations[Td, Axis, KeepDims], KeepOrReduceDims[S, Axis, KeepDims])]

Inherited methods

def callOp[T <: Supported, Tt <: TensorTypeDenotation, Td <: TensorShapeDenotation, S <: Shape](name: String, opName: String, inputs: Tuple, attrs: Map[String, Any])(using tt: ValueOf[Tt], td: TensorShapeDenotationOf[Td], s: ShapeOf[S]): Tensor[T, (Tt, Td, S)]
Inherited from:
Operator