在pytorch1.6中,必须如下调用
x = torch::reflection_pad2d(x, {1, 1, 1, 1});
不能用
x = torch::reflection_pad2d(x, {1});
或者
x = torch::reflection_pad2d(x, 1);
这是与python版本不同的地方,或许后面版本会改过来。
在pytorch1.6中,必须如下调用
x = torch::reflection_pad2d(x, {1, 1, 1, 1});
不能用
x = torch::reflection_pad2d(x, {1});
或者
x = torch::reflection_pad2d(x, 1);
这是与python版本不同的地方,或许后面版本会改过来。