RelationalReasoning Class Reference

Inheritance diagram for RelationalReasoning:

Public Member Functions
def	__init__ (self, num_layers, hidden_dim, **block_args)
	Constructor.
def	forward (self, x)
	forward
def	get_attention_maps (self, x)
	get attention maps

Public Attributes
	layers
	layers
	norm
	normalization

Detailed Description

RelationalReasoning

Definition at line 959 of file ieagan.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			num_layers,
			hidden_dim,
		**	block_args
	)

Constructor.

Definition at line 963 of file ieagan.py.

    def __init__(self, num_layers, hidden_dim, **block_args):
        super().__init__()
        
        self.layers = nn.ModuleList(
            [EncoderBlock(**block_args) for _ in range(num_layers)]
        )
        
        self.norm = nn.LayerNorm(hidden_dim)
 

Member Function Documentation

forward()

def forward	(		self,
			x
	)

forward

Definition at line 973 of file ieagan.py.

    def forward(self, x):
        for layer in self.layers:
            x = layer(x)
 
        x = self.norm(x)
        return x
 

get_attention_maps()

def get_attention_maps	(		self,
			x
	)

get attention maps

Definition at line 981 of file ieagan.py.

    def get_attention_maps(self, x):
        attention_maps = []
        for layer in self.layers:
            _, attn_map = layer.self_attn(x, return_attention=True)
            attention_maps.append(attn_map)
            x = layer(x)
        return attention_maps
 
 

Member Data Documentation

layers

layers

layers

Definition at line 966 of file ieagan.py.

norm

norm

normalization

Definition at line 970 of file ieagan.py.

---

The documentation for this class was generated from the following file:

• pxd/scripts/pxd/background_generator/models/ieagan.py