예제 #1
0
 def check_backward(self, c_prev1_data, c_prev2_data, x1_data, x2_data,
                    c_grad, h_grad):
     gradient_check.check_backward(
         functions.SLSTM(),
         (c_prev1_data, c_prev2_data, x1_data, x2_data),
         (c_grad, h_grad), dtype=numpy.float64,
         **self.check_backward_options)
예제 #2
0
 def check_backward(self, c_prev1_data, c_prev2_data, x1_data, x2_data,
                    c_grad, h_grad):
     gradient_check.check_backward(
         functions.SLSTM(),
         (c_prev1_data, c_prev2_data, x1_data, x2_data),
         (c_grad, h_grad), eps=1e-2)
예제 #3
0
    def __init__(self, n_children, n_units, n_label, classes=None):
        super().__init__(n_children, n_units, n_label, classes)

        self.add_link("slstm", F.SLSTM(self.n_children, n_units, n_units))