Dragon API Dragon v0.3.0 Python ArgMax¶ class dragon.vm.caffe.core.layers.ArgMax(layer_param)[source]¶ Compute the index of maximum elements along the given axis. Examples: layer { type: "ArgMax" bottom: "ip2" top: "cls" argmax_param { axis: 1 } }