Normally we can create a dropout layer by
self.drop_out = nn.Dropout(p=0.5)
For my case, the dropout probability will depend on the input and will be passed in the input vector x during forward propagation. Can anyone tell me how to achieve it?