:param d_model: :param d_k: :param d_v: :param sequence_length: :param h: :param batch_size: :param embedded_words: shape:[batch_size,sequence_length,embed_size]
(self,d_model,d_k,d_v,sequence_length,h,batch_size,num_layer=6,type='encoder',decoder_sent_length=None)
| 8 | base class has some common fields and functions. |
| 9 | """ |
| 10 | def __init__(self,d_model,d_k,d_v,sequence_length,h,batch_size,num_layer=6,type='encoder',decoder_sent_length=None): |
| 11 | """ |
| 12 | :param d_model: |
| 13 | :param d_k: |
| 14 | :param d_v: |
| 15 | :param sequence_length: |
| 16 | :param h: |
| 17 | :param batch_size: |
| 18 | :param embedded_words: shape:[batch_size,sequence_length,embed_size] |
| 19 | """ |
| 20 | self.d_model=d_model |
| 21 | self.d_k=d_k |
| 22 | self.d_v=d_v |
| 23 | self.sequence_length=sequence_length |
| 24 | self.h=h |
| 25 | self.num_layer=num_layer |
| 26 | self.batch_size=batch_size |
| 27 | self.type=type |
| 28 | self.decoder_sent_length=decoder_sent_length |
| 29 | |
| 30 | def sub_layer_postion_wise_feed_forward(self ,x ,layer_index,type) :# COMMON FUNCTION |
| 31 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected