Is there any layer or module in pytorch which works like tensorflow’s space_to_depth() 138 function? I have not found any concrete implementation of this operation neither in pytorch nor tensorflow! Could you please give me some hints about the implementation?