torchpack.nn.functional package

Submodules

torchpack.nn.functional.index module

torchpack.nn.functional.index.batched_index_select(inputs: torch.Tensor, indices: torch.Tensor, dim: int) → torch.Tensor

Module contents