RelationalReasoning Class Reference

Inheritance diagram for RelationalReasoning:

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

Public Attributes
	layers
	layers
	norm = nn.LayerNorm(hidden_dim)
	normalization

Detailed Description

RelationalReasoning

Definition at line 987 of file ieagan.py.

Constructor & Destructor Documentation

__init__()

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

Constructor.

Definition at line 991 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()

forward	(		self,
			x )

forward

Definition at line 1001 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()

get_attention_maps	(		self,
			x )

get attention maps

Definition at line 1009 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

Initial value:

=  nn.ModuleList(
            [EncoderBlock(**block_args) for _ in range(num_layers)]
        )

layers

Definition at line 994 of file ieagan.py.

norm

norm = nn.LayerNorm(hidden_dim)

normalization

Definition at line 998 of file ieagan.py.

---

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

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