What is the equivalent of tf.unstack()?

torch.unbind http://pytorch.org/docs/master/torch.html?highlight=unbind#torch.unbind

3 Likes