Skip to content

Commit 5be0d4e

Browse files
youge325youge325
authored
add squeeze and unsqueeze API test (#17)
1 parent d2e83a9 commit 5be0d4e

File tree

2 files changed

+205
-0
lines changed

2 files changed

+205
-0
lines changed

‎test/ops/SqueezeTest.cpp‎

Lines changed: 96 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,96 @@
1+
#include <ATen/ATen.h>
2+
#include <ATen/core/Tensor.h>
3+
#include <ATen/ops/ones.h>
4+
#include <gtest/gtest.h>
5+
6+
#include <string>
7+
#include <vector>
8+
9+
#include "../../src/file_manager.h"
10+
11+
extern paddle_api_test::ThreadSafeParam g_custom_param;
12+
13+
namespace at {
14+
namespace test {
15+
16+
using paddle_api_test::FileManerger;
17+
using paddle_api_test::ThreadSafeParam;
18+
19+
class SqueezeTest : public ::testing::Test {
20+
protected:
21+
void SetUp() override {
22+
// 创建一个包含大小为1的维度的tensor: shape = {2, 1, 3, 1, 4}
23+
tensor_with_ones = at::ones({2, 1, 3, 1, 4}, at::kFloat);
24+
}
25+
at::Tensor tensor_with_ones;
26+
};
27+
28+
// 测试 squeeze - 移除所有大小为1的维度
29+
TEST_F(SqueezeTest, SqueezeAll) {
30+
auto file_name = g_custom_param.get();
31+
FileManerger file(file_name);
32+
file.createFile();
33+
at::Tensor squeezed = tensor_with_ones.squeeze();
34+
file << std::to_string(squeezed.dim()) << " ";
35+
file << std::to_string(squeezed.numel()) << " ";
36+
for (int64_t i = 0; i < squeezed.dim(); ++i) {
37+
file << std::to_string(squeezed.sizes()[i]) << " ";
38+
}
39+
file.saveFile();
40+
}
41+
42+
// 测试 squeeze - 移除指定维度
43+
TEST_F(SqueezeTest, SqueezeDim) {
44+
auto file_name = g_custom_param.get();
45+
FileManerger file(file_name);
46+
file.createFile();
47+
// 移除维度1（大小为1）
48+
at::Tensor squeezed_dim1 = tensor_with_ones.squeeze(1);
49+
file << std::to_string(squeezed_dim1.dim()) << " ";
50+
file << std::to_string(squeezed_dim1.numel()) << " ";
51+
for (int64_t i = 0; i < squeezed_dim1.dim(); ++i) {
52+
file << std::to_string(squeezed_dim1.sizes()[i]) << " ";
53+
}
54+
file.saveFile();
55+
}
56+
57+
// 测试 squeeze_ - 原位移除所有大小为1的维度
58+
TEST_F(SqueezeTest, SqueezeInplaceAll) {
59+
auto file_name = g_custom_param.get();
60+
FileManerger file(file_name);
61+
file.createFile();
62+
// 记录原始数据指针
63+
void* original_ptr = tensor_with_ones.data_ptr();
64+
// 原位移除所有大小为1的维度
65+
tensor_with_ones.squeeze_();
66+
file << std::to_string(tensor_with_ones.dim()) << " ";
67+
file << std::to_string(tensor_with_ones.numel()) << " ";
68+
for (int64_t i = 0; i < tensor_with_ones.dim(); ++i) {
69+
file << std::to_string(tensor_with_ones.sizes()[i]) << " ";
70+
}
71+
// 验证是原位操作（数据指针未改变）
72+
file << std::to_string(tensor_with_ones.data_ptr() == original_ptr) << " ";
73+
file.saveFile();
74+
}
75+
76+
// 测试 squeeze_ - 原位移除指定维度
77+
TEST_F(SqueezeTest, SqueezeInplaceDim) {
78+
auto file_name = g_custom_param.get();
79+
FileManerger file(file_name);
80+
file.createFile();
81+
// 记录原始数据指针
82+
void* original_ptr = tensor_with_ones.data_ptr();
83+
// 原位移除维度1
84+
tensor_with_ones.squeeze_(1);
85+
file << std::to_string(tensor_with_ones.dim()) << " ";
86+
file << std::to_string(tensor_with_ones.numel()) << " ";
87+
for (int64_t i = 0; i < tensor_with_ones.dim(); ++i) {
88+
file << std::to_string(tensor_with_ones.sizes()[i]) << " ";
89+
}
90+
// 验证是原位操作（数据指针未改变）
91+
file << std::to_string(tensor_with_ones.data_ptr() == original_ptr) << " ";
92+
file.saveFile();
93+
}
94+
95+
} // namespace test
96+
} // namespace at

‎test/ops/UnsqueezeTest.cpp‎

Lines changed: 109 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,109 @@
1+
#include <ATen/ATen.h>
2+
#include <ATen/core/Tensor.h>
3+
#include <ATen/ops/ones.h>
4+
#include <gtest/gtest.h>
5+
6+
#include <string>
7+
#include <vector>
8+
9+
#include "../../src/file_manager.h"
10+
11+
extern paddle_api_test::ThreadSafeParam g_custom_param;
12+
13+
namespace at {
14+
namespace test {
15+
16+
using paddle_api_test::FileManerger;
17+
using paddle_api_test::ThreadSafeParam;
18+
19+
class UnsqueezeTest : public ::testing::Test {
20+
protected:
21+
void SetUp() override {
22+
// 创建一个基础tensor: shape = {2, 3, 4}
23+
tensor = at::ones({2, 3, 4}, at::kFloat);
24+
}
25+
at::Tensor tensor;
26+
};
27+
28+
// 测试 unsqueeze - 在维度0之前添加维度
29+
TEST_F(UnsqueezeTest, UnsqueezeDim0) {
30+
auto file_name = g_custom_param.get();
31+
FileManerger file(file_name);
32+
file.createFile();
33+
at::Tensor unsqueezed0 = tensor.unsqueeze(0);
34+
file << std::to_string(unsqueezed0.dim()) << " ";
35+
file << std::to_string(unsqueezed0.numel()) << " ";
36+
for (int64_t i = 0; i < unsqueezed0.dim(); ++i) {
37+
file << std::to_string(unsqueezed0.sizes()[i]) << " ";
38+
}
39+
file.saveFile();
40+
}
41+
42+
// 测试 unsqueeze - 在维度2之前添加维度
43+
TEST_F(UnsqueezeTest, UnsqueezeDim2) {
44+
auto file_name = g_custom_param.get();
45+
FileManerger file(file_name);
46+
file.createFile();
47+
at::Tensor unsqueezed2 = tensor.unsqueeze(2);
48+
file << std::to_string(unsqueezed2.dim()) << " ";
49+
file << std::to_string(unsqueezed2.numel()) << " ";
50+
for (int64_t i = 0; i < unsqueezed2.dim(); ++i) {
51+
file << std::to_string(unsqueezed2.sizes()[i]) << " ";
52+
}
53+
file.saveFile();
54+
}
55+
56+
// 测试 unsqueeze - 使用负索引在最后添加维度
57+
TEST_F(UnsqueezeTest, UnsqueezeNegativeDim) {
58+
auto file_name = g_custom_param.get();
59+
FileManerger file(file_name);
60+
file.createFile();
61+
at::Tensor unsqueezed_last = tensor.unsqueeze(-1);
62+
file << std::to_string(unsqueezed_last.dim()) << " ";
63+
file << std::to_string(unsqueezed_last.numel()) << " ";
64+
for (int64_t i = 0; i < unsqueezed_last.dim(); ++i) {
65+
file << std::to_string(unsqueezed_last.sizes()[i]) << " ";
66+
}
67+
file.saveFile();
68+
}
69+
70+
// 测试 unsqueeze_ - 原位在维度0之前添加维度
71+
TEST_F(UnsqueezeTest, UnsqueezeInplaceDim0) {
72+
auto file_name = g_custom_param.get();
73+
FileManerger file(file_name);
74+
file.createFile();
75+
// 记录原始数据指针
76+
void* original_ptr = tensor.data_ptr();
77+
// 原位在维度0之前添加维度
78+
tensor.unsqueeze_(0);
79+
file << std::to_string(tensor.dim()) << " ";
80+
file << std::to_string(tensor.numel()) << " ";
81+
for (int64_t i = 0; i < tensor.dim(); ++i) {
82+
file << std::to_string(tensor.sizes()[i]) << " ";
83+
}
84+
// 验证是原位操作（数据指针未改变）
85+
file << std::to_string(tensor.data_ptr() == original_ptr) << " ";
86+
file.saveFile();
87+
}
88+
89+
// 测试 unsqueeze_ - 原位使用负索引添加维度
90+
TEST_F(UnsqueezeTest, UnsqueezeInplaceNegativeDim) {
91+
auto file_name = g_custom_param.get();
92+
FileManerger file(file_name);
93+
file.createFile();
94+
// 记录原始数据指针
95+
void* original_ptr = tensor.data_ptr();
96+
// 原位在最后添加维度
97+
tensor.unsqueeze_(-1);
98+
file << std::to_string(tensor.dim()) << " ";
99+
file << std::to_string(tensor.numel()) << " ";
100+
for (int64_t i = 0; i < tensor.dim(); ++i) {
101+
file << std::to_string(tensor.sizes()[i]) << " ";
102+
}
103+
// 验证是原位操作（数据指针未改变）
104+
file << std::to_string(tensor.data_ptr() == original_ptr) << " ";
105+
file.saveFile();
106+
}
107+
108+
} // namespace test
109+
} // namespace at

0 commit comments

Comments
 (0)