コード例 #1
0
 def __init__(self, num_hidden, w_init=XavierUniform(), b_init=Zeros()):
     super().__init__()
     self.n_h = num_hidden
     self.initializers = {
         "W_g": w_init,
         "W_c": w_init,
         "b_g": b_init,
         "b_c": b_init
     }
コード例 #2
0
 def __init__(self,
              kernel,
              stride=(1, 1),
              padding="SAME",
              w_init=XavierUniform(),
              b_init=Zeros()):
     super().__init__(kernel, stride, padding, w_init, b_init)
     self.origin_stride = stride
     self.stride = (1, 1)
コード例 #3
0
ファイル: layer.py プロジェクト: yuqZY/tinynn
    def __init__(self,
                 num_out,
                 w_init=XavierUniform(),
                 b_init=Zeros()):
        super().__init__()

        self.initializers = {"w": w_init, "b": b_init}
        self.shapes = {"w": [None, num_out], "b": [num_out]}

        self.inputs = None
コード例 #4
0
ファイル: layer.py プロジェクト: yuqZY/tinynn
    def __init__(self, 
                 num_hidden,
                 activation,
                 bptt_trunc=None,
                 w_init=XavierUniform(),
                 b_init=Zeros()):
        super().__init__()
        self.num_hidden = num_hidden
        self.activation = activation
        self.bptt_trunc = bptt_trunc

        self.initializer = {"W": w_init, "V": w_init, "U": w_init,
                            "b": b_init, "c": b_init}
コード例 #5
0
    def __init__(self,
                 kernel,
                 stride=(1, 1),
                 padding="SAME",
                 w_init=XavierUniform(),
                 b_init=Zeros()):
        super().__init__()

        self.kernel_shape = kernel
        self.stride = stride
        self.initializers = {"w": w_init, "b": b_init}
        self.shapes = {"w": self.kernel_shape, "b": self.kernel_shape[-1]}

        self.padding_mode = padding
        self.padding = None