Is there a similar function in pytorch equivalent to tensorflow's space_to_batch_nd?

Is there a similar function in pytorch equivalent to tensorflow’s space_to_batch_nd?

2 Likes