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 padding to avg pooling

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-26 18:54:12 +02:00
parent d1fc45d9e0
commit 4a67b708f0
6 changed files with 60 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tools/pooling_test.py
View File

@@ -103,6 +103,15 @@ def gen_avg_pool_non_square_stride_test_result():
print_cpp_vector(output)
def gen_avg_pool_non_square_padding_test_result():
input = _get_pool_input()
output = torch.nn.AvgPool2d(kernel_size=2, stride=2, padding=(1, 0))(input)
output = torch.flatten(output)
print_cpp_vector(output)
if __name__ == "__main__":
print("Generating test results...")
@@ -125,3 +134,5 @@ if __name__ == "__main__":
gen_avg_non_square_pool_test_result()
print("Avg pool non square stride test:")
gen_avg_pool_non_square_stride_test_result()
print("Avg pool non square padding test:")
gen_avg_pool_non_square_padding_test_result()