mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
14 lines
316 B
C++
14 lines
316 B
C++
|
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
#include <torch/special.h>
|
||
|
|
#include <torch/torch.h>
|
||
|
|
|
||
|
|
#include <test/cpp/api/support.h>
|
||
|
|
|
||
|
|
// Simple test that verifies the special namespace is registered properly
|
||
|
|
// properly in C++
|
||
|
|
TEST(SpecialTest, special) {
|
||
|
|
auto t = torch::randn(128, torch::kDouble);
|
||
|
|
torch::special::gammaln(t);
|
||
|
|
}
|