matchzoo.modules.semantic_composite
¶
Semantic composite module for DIIN model.
Module Contents¶
Classes¶
SemanticComposite module. |
-
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.