def __init__(self, n_layers, in_size, out_size, dropout=0.5,
initialW=None, initial_bias=None):
if int(chainer_version[0]) > 2:
super(GRU, self).__init__(n_layers, in_size, out_size, dropout,
initialW, initial_bias)
else:
super(GRU, self).__init__(n_layers, in_size, out_size, dropout)
评论列表
文章目录