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

Getting Started

  • Installation
  • Introduction

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
Edit this page

torchsparse.nn.utils#

fapply(input: SparseTensor, fn: Callable[[...], Tensor], *args, **kwargs) → SparseTensor[source]#
get_kernel_offsets(size: Union[int, Tuple[int, ...]], stride: Union[int, Tuple[int, ...]] = 1, dilation: Union[int, Tuple[int, ...]] = 1, device: str = 'cpu') → Tensor[source]#
Next
torchsparse.utils
Previous
torchsparse.nn.modules
Copyright © 2022, MIT HAN Lab
Made with Sphinx and @pradyunsg's Furo
On this page
  • torchsparse.nn.utils
    • fapply()
    • get_kernel_offsets()