title: Is there an equivalent PyTorch function for `tf.nn.space_to_depth`
body: <p>As the title says, is there an equivalent PyTorch function for <a href="https://www.tensorflow.org/api_docs/python/tf/nn/space_to_depth" rel="nofollow noreferrer">tf.nn.space_to_depth</a>?</p>