Table of Contents

Shortcuts

Reversible layer¶

class xformers.components.reversible.Deterministic(net: Module)[source]¶

record_rng(*args)[source]¶

forward(*args, record_rng: bool = False, set_rng: bool = False, **kwargs)[source]¶

training: bool¶

class xformers.components.reversible.ReversibleBlock(f: Module, g: Module, split_dim: int = - 1)[source]¶

forward(x: Tensor, f_args={}, g_args={})[source]¶

backward_pass(y: Tensor, dy: Tensor, f_args={}, g_args={})[source]¶

training: bool¶

class xformers.components.reversible.ReversibleSequence(blocks: ModuleList)[source]¶

forward(x, arg_route=(True, False), **kwargs)[source]¶

training: bool¶