Module `ai.utils.torch`

Contains general PyTorch utility methods.

Expand source code

"""Contains general PyTorch utility methods."""


from . import nn, random


__all__ = ["nn", "random"]

Sub-modules

ai.utils.torch.nn: Utility methods and classes for neural networks defined in PyTorch.
ai.utils.torch.random: Utility methods dealing with random numbers in PyTorch.