matchzoo.modules.semantic_composite

Semantic composite module for DIIN model.

Module Contents

class matchzoo.modules.semantic_composite.SemanticComposite(in_features, dropout_rate: float = 0.0)

Bases: torch.nn.Module

SemanticComposite module.

Apply a self-attention layer and a semantic composite fuse gate to compute the encoding result of one tensor.

Parameters:

• in_features – Feature size of input.
• dropout_rate – The dropout rate.

Examples

>>> import torch
>>> module = SemanticComposite(in_features=10)
>>> x = torch.randn(4, 5, 10)
>>> x.shape
torch.Size([4, 5, 10])
>>> module(x).shape
torch.Size([4, 5, 10])

forward(self, x)

Forward.