PyTorch function similar to tf.set_shape

vision
1

Dose PyTorch has something similar to tf.set_shape(). Link to the TF set_shape.

2

You can use the .view() function. view

1 Like
3

You can use either .view() or torch.reshape