示例#1
0
 static public TorchTensor Dropout(TorchTensor x, double probability = 0.5)
 {
     using (var d = Modules.Dropout(probability)) {
         return(d.Forward(x));
     }
 }
示例#2
0
 /// <summary>
 /// During training, randomly zeroes some of the elements of the input tensor with probability p using samples from a Bernoulli distribution.
 /// Each channel will be zeroed out independently on every forward call.
 /// </summary>
 /// <param name="x">Input tensor</param>
 /// <param name="probability">Probability of an element to be zeroed. Default: 0.5</param>
 /// <param name="inPlace">If set to true, will do this operation in-place. Default: false</param>
 /// <returns></returns>
 static public TorchTensor Dropout(TorchTensor x, double probability = 0.5, bool inPlace = false)
 {
     using (var d = Modules.Dropout(probability, inPlace)) {
         return(d.forward(x));
     }
 }