torchgear

torchgear provides utility extensions for PyTorch: tensor math helpers, shape utilities, and torch.nn.Module mixins for device/dtype consistency and parameter freezing.

User guide

• Getting started
  • Installation
  • Quick example

API reference

• torchgear.math
  • abs2()
  • expi()
  • polynomial()
  • ssqrt()
• torchgear.tensor
  • ShapeError
  • as1d()
  • broadcastable()
• torchgear.module
  • DeviceMixIn
  • DtypeMixIn
  • FreezeParamMixIn
  • TensorAsDelegate
  • TensorContainerMixIn
  • TorchgearModule
  • WrapperModule

Indices and tables

• Index

• Module Index

• Search Page