We currently have Dropout2d and Dropout3d, which does similar thing but for 2d and 3d inputs. We plan to support more general feature dropouts soon. In fact, you can see it mostly done in this PR, but was eventually blocked by something that needs to be fixed: https://github.com/pytorch/pytorch/pull/9008 355