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
  }
}