argMax(axis:keepDimension:)
Adds an argmax operation to the current graph.
Declaration
func argMax(axis: Int, keepDimension: Bool) -> BNNSGraph.Builder.Tensor<Int32>Discussion
This function calculates the first index of the maximum value in the tensor.