:param d_model: :param d_k: :param d_v: :param sequence_length: :param h: :param batch_size: :param Q: :param K_s: :param K_v_encoder: shape:[batch_size,sequence_length,embed_size]. it is the output from encoder
(self,d_model,d_k,d_v,sequence_length,h,batch_size,Q,K_s,K_v_encoder,decoder_sent_length,
num_layer=6,type='decoder',is_training=True,mask=None,dropout_keep_prob=None)
| 17 | |
| 18 | class Decoder(BaseClass): |
| 19 | def __init__(self,d_model,d_k,d_v,sequence_length,h,batch_size,Q,K_s,K_v_encoder,decoder_sent_length, |
| 20 | num_layer=6,type='decoder',is_training=True,mask=None,dropout_keep_prob=None): |
| 21 | """ |
| 22 | :param d_model: |
| 23 | :param d_k: |
| 24 | :param d_v: |
| 25 | :param sequence_length: |
| 26 | :param h: |
| 27 | :param batch_size: |
| 28 | :param Q: |
| 29 | :param K_s: |
| 30 | :param K_v_encoder: shape:[batch_size,sequence_length,embed_size]. it is the output from encoder |
| 31 | """ |
| 32 | super(Decoder, self).__init__(d_model, d_k, d_v, sequence_length, h, batch_size, num_layer=num_layer) |
| 33 | self.Q=Q |
| 34 | self.K_s=K_s |
| 35 | self.K_v_encoder=K_v_encoder |
| 36 | self.type=type |
| 37 | self.initializer = tf.random_normal_initializer(stddev=0.1) |
| 38 | self.is_training=is_training |
| 39 | self.decoder_sent_length=decoder_sent_length |
| 40 | self.mask=mask |
| 41 | self.dropout_keep_prob=dropout_keep_prob |
| 42 | |
| 43 | def decoder_fn(self): |
| 44 | start = time.time() |
nothing calls this directly
no outgoing calls
no test coverage detected