Add noise shape functionality to dropout
Benny-Nottonson opened this issue · 0 comments
Benny-Nottonson commented
Is your feature request related to a problem? Please describe.
Dropout should also have a noise shape option that allows for a mask to be used on top of the noise, this likely means it will need to be moved to conv and maxpool kernel file since it will rely on another tensor.
Describe the solution you'd like
Dropout should be rewritten to allow for the optional use of a noise mask.