pytorch

1
#include <gtest/gtest.h>
2

3
#include <torch/nested.h>
4
#include <torch/torch.h>
5

6
#include <test/cpp/api/support.h>
7

8
// Simple test that verifies the nested namespace is registered properly
9
//   properly in C++
10
TEST(NestedTest, Nested) {
11
  auto a = torch::randn({2, 3});
12
  auto b = torch::randn({4, 5});
13
  auto nt = torch::nested::nested_tensor({a, b});
14
  torch::nested::to_padded_tensor(nt, 0);
15
}
16