Argmax

public final class Argmax : Model

Argmax model.


                    
                    
                    init(axis:name:)

Swift

public init(axis: Int, name: String = "")


                    
                    
                    callAsFunction(_:_:streamContext:)

Swift

public func callAsFunction<T: DynamicGraph.TensorGroup>(
  _ left: T, _ right: T, streamContext: StreamContext? = nil
) -> T