[Hackathon 7th No.55] BUG FIX paddle.nn.functional.fold

[DrRyanHuang]

PR Category

User Experience

PR Types

Bug fixes

Description

import paddle

unfolded = paddle.randn([16, 4410, 201])
folded = paddle.nn.functional.fold(
    unfolded,
    output_sizes=(1, 445410),
    kernel_sizes=(1, 4410),
    strides=(1, 2205), )
print(folded.shape)

(InvalidArgument) The `output_height` should be greater than one, but received output_height: 1 .
  [Hint: Expected output_height > 1, but received output_height:1 <= 1:1.] (at /paddle/paddle/phi/infermeta/unary.cc:1597)
  File "/home/aistudio/PaddleSpeech/audio/audiotools/core/dsp.py", line 131, in overlap_and_add
    folded = paddle.nn.functional.fold(
  File "/home/aistudio/PaddleSpeech/audio/tests/audiotools/core/test_dsp.py", line 31, in _test
    recombined = windowed_signal.overlap_and_add(hop_duration)
  File "/home/aistudio/PaddleSpeech/audio/tests/audiotools/core/test_dsp.py", line 36, in test_overlap_add
    _test(nz_signal)
  File "/home/aistudio/PaddleSpeech/audio/tests/audiotools/core/test_dsp.py", line 39, in <module>
    test_overlap_add(10.0, 44100, 0.1)

However,

import torch
unfolded = torch.randn(16, 4410, 201)
folded = torch.nn.functional.fold(
    unfolded,
    output_size=(1, 445410),
    kernel_size=(1, 4410),
    stride=(1, 2205),
)
print(folded.shape) # torch.Size([16, 1, 1, 445410])

• 相关PR：[Hackathon 7th No.55] Add audiotools to PaddleSpeech PaddleSpeech#3900
• RFC：【Hackathon 7th No.55】在 PaddleSpeech 中实现 audiotools community#1017

[paddle-bot]

你的PR提交成功，感谢你对开源项目的贡献!
请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[luotao1]

CodeStyle没过