ArgMax

class dragon.vm.caffe.layers.ArgMax(layer_param)[source]

Compute the indices of maximum elements along the given axis.

Examples:

layer {
    type: "ArgMax"
    bottom: "ip2"
    top: "cls"
    argmax_param {
        top_k: 1
        axis: 1
    }
}