I wanted to implement a bayesian CNN from scratch using the PyTorch framework. I searched for the implementation of dropout in order to write a custom modification. I searched for the function given in the definition of dropout3d and could not find the apply() function?
Could anyone help me? Also I’m new to this forum. Thanks.
Link:http://pytorch.org/docs/master/_modules/torch/nn/functional.html#dropout3d 93
def dropout3d(input, p=0.5, training=False, inplace=False):
return _functions.dropout.FeatureDropout.apply(input, p, training, inplace)