Misc
Miscelleaneous handy functions.
Functions:
|
Compute the argmax of a tensor. |
- bardensr.misc.argmax_nd(x)
Compute the argmax of a tensor.
Input: x, an n-dimensional numpy array
Output: idx: an integer n-tuple indicating an index where x reaches its greatest point
Seems like this is the kind of think numpy might provide, but I think it isn’t.