1 | #include <gtest/gtest.h> |
---|---|
2 | |
3 | #include <torch/special.h> |
4 | #include <torch/torch.h> |
5 | |
6 | #include <test/cpp/api/support.h> |
7 | |
8 | // Simple test that verifies the special namespace is registered properly |
9 | // properly in C++ |
10 | TEST(SpecialTest, special) { |
11 | auto t = torch::randn(128, torch::kDouble); |
12 | torch::special::gammaln(t); |
13 | } |
14 |