matchzoo.modules.character_embedding
¶
Character embedding module.
Module Contents¶
-
class
matchzoo.modules.character_embedding.
CharacterEmbedding
(char_embedding_input_dim: int = 100, char_embedding_output_dim: int = 8, char_conv_filters: int = 100, char_conv_kernel_size: int = 5)¶ Bases:
torch.nn.Module
Character embedding module.
Parameters: - char_embedding_input_dim – The input dimension of character embedding layer.
- char_embedding_output_dim – The output dimension of character embedding layer.
- char_conv_filters – The filter size of character convolution layer.
- char_conv_kernel_size – The kernel size of character convolution layer.
Examples
>>> import torch >>> character_embedding = CharacterEmbedding() >>> x = torch.ones(10, 32, 16, dtype=torch.long) >>> x.shape torch.Size([10, 32, 16]) >>> character_embedding(x).shape torch.Size([10, 32, 100])
-
forward
(self, x)¶ Forward.