This class implements a layer that finds, for a single input, the coordinates of the maximum elements along a given dimension.
void SetDimension(TBlobDim d);
Sets the dimension along which the maximums should be found.
This layer has no trainable parameters.
The single input accepts a blob of any size.
The single output contains a blob with int
data of the dimensions:
- the
GetDimension()
dimension equals1
- the other dimensions are the same as in the input blob
The elements of the blob contain the coordinates of the maximums taken along the specified dimension.