frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implement col2im decomposition and fix im2col and add a few preconditions (#85541)

Browse Source 
As per title
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85541
Approved by: https://github.com/jansel
This commit is contained in:
lezcano
2022-09-29 18:19:57 +00:00
committed by PyTorch MergeBot
parent 1f38abb5d2
commit 787028cadb
7 changed files with 198 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/cpp/api/modules.cpp
View File

@ -1225,7 +1225,7 @@ TEST_F(ModulesTest, Unfold) {
model(torch::randn({1, 2, 2, 2})),
"Given input with spatial size (2, 2), kernel_size=(2, 3), "
"dilation=(1, 1), padding=(0, 0), calculated shape of the array of "
"sliding blocks as (1, 0), which is too small (non-positive).");
"sliding blocks as (1, 0), but its components must be at least one.");
}
}