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.