mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
e217b30b0f
First step towards #83775 - only `to_padded_tensor` is moved to the nested namespace for now - following the schema used for `special`, `fft`, `linalg` and other namespaces, nested functions are registered in native_functions.yaml as `nested_{function_name}` and are bound to the desired Python name in `torch/nested/__init__.py`, and the desired C++ name in `torch/csrc/api/include/torch/nested.h`. ~~**Question**: should we keep the documentation for `Tensor.to_padded_tensor` or can this deleted since it is shared by `torch.nested.to_padded_tensor`?~~ [generated nested docs](https://docs-preview.pytorch.org/84102/nested.html?highlight=nested#module-torch.nested) Differential Revision: [D39361148](https://our.internmc.facebook.com/intern/diff/D39361148) Pull Request resolved: https://github.com/pytorch/pytorch/pull/84102 Approved by: https://github.com/drisspg
16 lines
386 B
C++
16 lines
386 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <torch/nested.h>
|
|
#include <torch/torch.h>
|
|
|
|
#include <test/cpp/api/support.h>
|
|
|
|
// Simple test that verifies the nested namespace is registered properly
|
|
// properly in C++
|
|
TEST(NestedTest, Nested) {
|
|
auto a = torch::randn({2, 3});
|
|
auto b = torch::randn({4, 5});
|
|
auto nt = torch::nested_tensor({a, b});
|
|
torch::nested::to_padded_tensor(nt, 0);
|
|
}
|