vectorz
vectorz copied to clipboard
Equivalent to Numpy amin, argsort?
I'm looking to convert numpy code and was wondering where the equilavent functions/method as amin, argsort are... Would be great if you could point me to a Class or where the implementation is?
There isn't a precise equivalent at present.
Depending on what you want to do, it should be fairly easy to get the same behaviour. e.g. for the equivalent of amin
you should be able to iterate over the slices in a given dimension with INDArray.slice(axis,i)
and choose the one that has the minimum value according to slice.elementMin()
Most functions for general purpose arrays are available via the mikera.arrayz.INDArray
interface, this is often the best place to start looking.