Contents Menu Expand Light mode Dark mode Auto light/dark mode
TorchSparse documentation
TorchSparse documentation

Getting Started

  • Introduction
  • Installation
  • Basic Concepts
  • Changes in TorchSparse v2.1
  • Examples for TorchSparse v2.1

Guides

  • Enabling Inference Optimizations

Reference

  • API Reference
    • torchsparse
    • torchsparse.backbones
      • torchsparse.backbones.modules
    • torchsparse.nn
      • torchsparse.nn.functional
      • torchsparse.nn.modules
      • torchsparse.nn.utils
    • torchsparse.utils
  • Citation

Support

  • Frequently Asked Questions
Back to top

torchsparse.nn.utils#

fapply(input: SparseTensor, fn: Callable[[...], Tensor], *args, **kwargs) → SparseTensor[source]#
get_kernel_offsets(size: int | Tuple[int, ...], stride: int | Tuple[int, ...] = 1, dilation: int | Tuple[int, ...] = 1, device: str = 'cpu') → Tensor[source]#
Next
torchsparse.utils
Previous
torchsparse.nn.modules
Copyright © 2023, MIT HAN Lab. Document author: Xiuyu Li. Modified by: Haotian Tang and Shang Yang.
Made with Sphinx and @pradyunsg's Furo
On this page
  • torchsparse.nn.utils
    • fapply()
    • get_kernel_offsets()