add remove_node in Partition class (#47452)
Summary:
add remove_node method in Partition class for the future use
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47452
Reviewed By: glaringlee, gcatron
Differential Revision: D24762770
Pulled By: scottxu0730
fbshipit-source-id: 35473ab7322d8e6ecab1c2624b668342bfec4cca
