diff options
Diffstat (limited to 'src/caffe/proto/caffe.proto')
-rw-r--r-- | src/caffe/proto/caffe.proto | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/caffe/proto/caffe.proto b/src/caffe/proto/caffe.proto index f52c941b..a8747c12 100644 --- a/src/caffe/proto/caffe.proto +++ b/src/caffe/proto/caffe.proto @@ -443,6 +443,11 @@ message ArgMaxParameter { // If true produce pairs (argmax, maxval) optional bool out_max_val = 1 [default = false]; optional uint32 top_k = 2 [default = 1]; + // The axis along which to maximise -- may be negative to index from the + // end (e.g., -1 for the last axis). + // By default ArgMaxLayer maximizes over the flattened trailing dimensions + // for each index of the first / num dimension. + optional int32 axis = 3; } message ConcatParameter { |