Class: Chainer::Functions::Noise::DropoutGrad

Inherits:
Chainer::FunctionNode show all
Defined in:
lib/chainer/functions/noise/dropout.rb

Overview

Computes the gradient of the Dropout function.

Instance Attribute Summary

Attributes inherited from Chainer::FunctionNode

#inputs, #outputs, #rank

Instance Method Summary collapse

Methods inherited from Chainer::FunctionNode

#apply, #backward_accumulate, #forward_cpu, #get_retained_inputs, #get_retained_outputs, #label, #output_data, #retain_inputs, #retain_outputs, #unchain

Constructor Details

#initialize(mask) ⇒ DropoutGrad

Returns a new instance of DropoutGrad.



45
46
47
 
# File 'lib/chainer/functions/noise/dropout.rb', line 45

def initialize(mask)
  @mask = mask
end

Instance Method Details

#backward(indexes, gy) ⇒ Object 



54
55
56
 
# File 'lib/chainer/functions/noise/dropout.rb', line 54

def backward(indexes, gy)
  DropoutGrad.new(@mask).apply(gy)
end

#forward(inputs) ⇒ Object 



49
50
51
52
 
# File 'lib/chainer/functions/noise/dropout.rb', line 49

def forward(inputs)
  y = inputs.first * @mask
  [y]
end