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