ReversibleEmbedding¶
Embedding layer that can project backwards to the input dim.
Abstract Signature:
ReversibleEmbedding(input_dim: int, output_dim: int, tie_weights: bool = True)
PyTorch
API:
—Strategy: Plugin (reversible_embedding_shim)
Embedding layer that can project backwards to the input dim.
Abstract Signature:
ReversibleEmbedding(input_dim: int, output_dim: int, tie_weights: bool = True)
—