Commit Graph

3 Commits

Author SHA1 Message Date
Jack Zhou
cec0d502e0 Add softmax function (#93)
* Add softmax function

* Add softmax unittest

* Add Softmax docs

* Add function directory

* Add comment for FD_VISIT_ALL_TYPES macro
2022-08-11 12:01:19 +08:00
Jack Zhou
7fb8dd7916 Add Transpose function (#91)
* Add Transpose function

* csrcs->csrc

* Add transpose unittest

* Add reduce_max_large_dim unittest
2022-08-10 19:00:16 +08:00
Jack Zhou
c7d37b6732 Add reduce functions for FDTensor (#81)
* Add eigen tensor structure

* Add Reduce function

* Add cpp unittest framework

* Add reduce function unittest

* Add pr template

* Add comments and docs for reduce function

* Fix typo

* Add ENABLE_FDTENSOR_FUNC macro

* Add Todo comment

* Add CheckData overload

* Fix CheckData overload operator()

Co-authored-by: Jason <jiangjiajun@baidu.com>
2022-08-10 10:31:04 +08:00