iesl / box-embeddings

Box Embeddings as Modules
https://www.iesl.cs.umass.edu/box-embeddings
Apache License 2.0
100 stars 10 forks source link

Add center, offset, side-length and minus_Z to all box tensors #20

Open mboratko opened 3 years ago

mboratko commented 3 years ago

This would be added to the base class, where we can perform it generically using zZ, but certain parameterizations may override this with more efficient implementations