Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add adaptive avg pooling

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-30 17:17:31 +02:00
parent 9faf20876a
commit 8168f02f58
5 changed files with 143 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tools/pooling_test.py
View File

@@ -126,6 +126,15 @@ def gen_avg_pool_non_square_padding_test_result():
print_cpp_vector(output)
def gen_adaptive_avg_pool_test_result():
input = _get_pool_input()
output = torch.nn.AdaptiveAvgPool2d((2, 2))(input)
output = torch.flatten(output)
print_cpp_vector(output)
if __name__ == "__main__":
print("Generating test results...")
@@ -152,3 +161,8 @@ if __name__ == "__main__":
gen_avg_pool_non_square_stride_test_result()
print("Avg pool non square padding test:")
gen_avg_pool_non_square_padding_test_result()
print("--------------")
print("Adaptive avg pool test:")
gen_adaptive_avg_pool_test_result()