mindtext.modules.decoder.norm_decoder

class mindtext.modules.decoder.norm_decoder.NormalDecoder (num_filters: int, num_classes: int, classes_dropout: float = 0.1, activation: str = ‘relu’)

使用一个dropout层和一个全连接层(dense layer)来进行解码。

Example

>>> 

init (num_filters: int, num_classes: int, classes_dropout: float = 0.1, activation: str = ‘relu’)

参数

  • num_filters (int): 过滤器的数量,默认为256。

  • num_classes (int): 输出分类的数量。

  • classes_dropout (float): 对词嵌入进行dropout,默认为0.1。

  • activation (str): 激活函数,默认为relu。

construct (x: mindspore.Tensor)

使用解码层进行解码

参数

  • x (mindspore.Tensor):输入的向量。

返回

  • x(mindspore.Tensor):解码器最后输出表示。