def __init__(self, input_size, hidden_size, num_embeddings=128, CUDA=True): super(AttentionCell, self).__init__() self.i2h = nn.Linear(input_size, hidden_size, bias=False) self.h2h = nn.Linear(hidden_size, hidden_size) self.score = nn.Linear(hidden_size, 1, bias=False) self.rnn = nn.GRUCell(input_size + num_embeddings, hidden_size) self.hidden_size = hidden_size self.input_size = input_size self.num_embeddings = num_embeddings self.fracPickup = fracPickup(CUDA=CUDA)
def __init__(self, input_size, hidden_size, num_embeddings=128, CUDA=True): """ :param input_size: 输入维度 :param hidden_size: cell个数 :param num_embeddings: :param CUDA: GPU支持 """ super(AttentionCell, self).__init__() self.i2h = nn.Linear(input_size, hidden_size, bias=False) self.h2h = nn.Linear(hidden_size, hidden_size) self.score = nn.Linear(hidden_size, 1, bias=False) self.rnn = nn.GRUCell(input_size + num_embeddings, hidden_size) self.hidden_size = hidden_size self.input_size = input_size self.num_embeddings = num_embeddings self.fracPickup = fracPickup(CUDA=CUDA) # 论文中的FP方法,提高识别准确率