box_embeddings.modules.volume.volume

Module Contents

class Volume(log_scale: bool = True, volume_temperature: float = 0.0, intersection_temperature: float = 0.0, **kwargs: Any)

Bases: box_embeddings.modules.volume._volume._Volume

One for All volume class

forward(self, box_tensor: box_embeddings.parameterizations.box_tensor.BoxTensor) → torch.Tensor

Base implementation is hard (ReLU) volume.

Parameters

box_tensor – Input box tensor

Returns

torch.Tensor