returnn.frontend._native

Native code as Python extension module for the RETURNN frontend, including tensor methods and ops.

returnn.frontend._native.get_module(*, verbose: bool = False)[source]
Returns:

native Python extension module

returnn.frontend._native.setup()[source]

Setup the native code.

returnn.frontend._native.setup_torch()[source]

Like setup(), but specifically for the PyTorch backend. This assumes that we can import torch, unlike setup().