pytorch上space2depth的实现

申明：转载注明出处https://www.cnblogs.com/wioponsen/p/14570312.html方式一 使用torch.nn.functional.unfolddef space_