pull/3900/head
drryanhuang 9 months ago
parent 7eb7ec2ffe
commit 35ee7da5fd

@ -44,8 +44,8 @@ def _unfold1d(x, kernel_size, stride):
unfolded = paddle.nn.functional.unfold( unfolded = paddle.nn.functional.unfold(
x, x,
kernel_sizes=(kernel_size, 1), kernel_sizes=[kernel_size, 1],
strides=(stride, 1), ) strides=[stride, 1], )
unfolded = unfolded.transpose([0, 2, 1]) unfolded = unfolded.transpose([0, 2, 1])
unfolded = unfolded.reshape([N, C, *unfolded.shape[1:]]) unfolded = unfolded.reshape([N, C, *unfolded.shape[1:]])

Loading…
Cancel
Save