pyabsa.networks.squeeze_embedding¶
Classes¶
Squeeze sequence embedding length to the longest one in the batch |
Module Contents¶
- class pyabsa.networks.squeeze_embedding.SqueezeEmbedding(batch_first=True)¶
Bases:
torch.nn.ModuleSqueeze sequence embedding length to the longest one in the batch
- batch_first = True¶
- forward(x, x_len)¶
sequence -> sort -> pad and pack -> unpack ->unsort :param x: sequence embedding vectors :param x_len: numpy/tensor list :return: