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 }
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)
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
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}
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