earthkit.utils.array package

Subpackages

• array.converter
  • Submodules
    • array.converter.cupy
    • array.converter.jax
    • array.converter.numpy
    • array.converter.torch
    • array.converter.unknown
  • Module contents
    • earthkit.utils.array.converter.FromNumpyConverter
      • FromNumpyConverter
    • earthkit.utils.array.converter.FromCupyConverter
      • FromCupyConverter
    • earthkit.utils.array.converter.FromTorchConverter
      • FromTorchConverter
    • earthkit.utils.array.converter.FromJaxConverter
      • FromJaxConverter
    • earthkit.utils.array.converter.FromUnknownConverter
      • FromUnknownConverter
• array.namespace
  • Submodules
    • array.namespace.cupy
    • array.namespace.jax
    • array.namespace.numpy
    • array.namespace.torch
    • array.namespace.unknown
  • Module contents
    • earthkit.utils.array.namespace.PatchedNumpyNamespace
      • PatchedNumpyNamespace
    • earthkit.utils.array.namespace.PatchedCupyNamespace
      • PatchedCupyNamespace
    • earthkit.utils.array.namespace.PatchedTorchNamespace
      • PatchedTorchNamespace
    • earthkit.utils.array.namespace.PatchedJaxNamespace
      • PatchedJaxNamespace
    • earthkit.utils.array.namespace.UnknownPatchedNamespace
      • UnknownPatchedNamespace
• array.testing
  • Submodules
    • array.testing.testing
  • Module contents
    • earthkit.utils.array.testing.NAMESPACE_DEVICES
      • NAMESPACE_DEVICES

Submodules

• array.array_namespace
• array.convert

Module contents

array_namespace(*args)	Return the array namespace of the arguments.
convert(array, *[, device, array_namespace])	Return a copy/view of a converted array.