Jack Zhou b6247238f5 Add argmax, argmin function (#104) ...

* Add argmax argmin function

* Add unittest for argmax, argmin

2022-08-12 20:22:11 +08:00

..

eigen.cc

add csrc code

2022-08-10 02:52:36 +00:00

eigen.h

Upgrade FDASSERT (#97)

2022-08-11 19:42:31 +08:00

reduce_functor.h

add csrc code

2022-08-10 02:52:36 +00:00

reduce.cc

Add argmax, argmin function (#104)

2022-08-12 20:22:11 +08:00

reduce.h

Add argmax, argmin function (#104)

2022-08-12 20:22:11 +08:00

softmax.cc

Upgrade FDASSERT (#97)

2022-08-11 19:42:31 +08:00

softmax.h

Add softmax function (#93)

2022-08-11 12:01:19 +08:00

transpose.cc

Upgrade FDASSERT (#97)

2022-08-11 19:42:31 +08:00

transpose.h

Add Transpose function (#91)

2022-08-10 19:00:16 +08:00