ini_type: initialization type for weight parameters: plain, xavier, or he
'''
def __init__(self, input_shape, n_out, ini_type="plain"):
ini_type: initialization type for weight parameters: plain, xavier, or he
'''
def __init__(self, input_shape, n_out, ini_type="plain"):