mipcandy.data.geometric

Module Contents

Functions

ensure_num_dimensions
orthographic_views
aggregate_orthographic_views
crop

API

mipcandy.data.geometric.ensure_num_dimensions(x: torch.Tensor, num_dimensions: int, *, append_before: bool = True) → torch.Tensor

mipcandy.data.geometric.orthographic_views(x: torch.Tensor, reduction: Literal[mean, sum] = 'mean') → tuple[torch.Tensor, torch.Tensor, torch.Tensor]

mipcandy.data.geometric.aggregate_orthographic_views(d: torch.Tensor, h: torch.Tensor, w: torch.Tensor) → torch.Tensor

mipcandy.data.geometric.crop(t: torch.Tensor, bbox: Sequence[int]) → torch.Tensor