is_local_max
is_local_max(matrix: any[] | Mat | Tensor, dim: number) : Mat | Tensor
param matrix
- an Array
, Mat
or Tensor
, i.e. the construct to find the local maximum value of the input.
param dim
- '0' (row) or '1' (column) - the direction to go across in finding the local maxima
returns Mat | Tensor
- a Mat
or Tensor
with '1' denoting the local max value is there and '0' denoting that it is not a local maxima.
Note that if dim
is '1', then matrix
must be at least a 2-D input.
This function finds all the local maximum value locations of the input.