pyabsa.networks.squeeze_embedding

Module Contents

Classes

SqueezeEmbedding

Squeeze sequence embedding length to the longest one in the batch

class pyabsa.networks.squeeze_embedding.SqueezeEmbedding(batch_first=True)

Bases: torch.nn.Module

Squeeze sequence embedding length to the longest one in the batch

forward(x, x_len)

sequence -> sort -> pad and pack -> unpack ->unsort :param x: sequence embedding vectors :param x_len: numpy/tensor list :return: