delu.nn

An extension to torch.nn.

Lambda	A wrapper for functions from torch and methods of torch.Tensor.
NLinear	N separate linear layers for N embeddings: (*, *N, D1) -> (*, *N, D2).