Quansight-Labs/numpy.net

argmax speed up

Closed this issue · 1 comments

williamlzw commented 
var time0 = DateTime.Now;
var argmaxArr = encoderOutArr.ArgMax(-1);
var time1 = DateTime.Now;
Console.WriteLine((time1 - time0).TotalMilliseconds);

encoderOutArr.shape [97,25055]
cost 100ms.

torchsharp 0.105.0 cpu version

var time0 = DateTime.Now;
var argmaxArr = encoderOutArr.argmax(-1);
var time1 = DateTime.Now;
Console.WriteLine((time1 - time0).TotalMilliseconds);

cost time 3ms
Suggest referring to torchsharp's argmax implementation to speed up this method.