HybridBlock.hybridize

HybridBlock.hybridize(self, active=True, **kwargs)

Activates or deactivates HybridBlock s recursively. Has no effect on non-hybrid children.

Parameters:

active : bool, default True

Whether to turn hybrid on or off.

static_alloc : bool, default False

Statically allocate memory to improve speed. Memory usage may increase.

static_shape : bool, default False

Optimize for invariant input shapes between iterations. Must also set static_alloc to True. Change of input shapes is still allowed but slower.