class

GS::MASt3R::MASt3REncoderBlock

Inherits Reference < Object

MASt3R encoder block with optional cross-attention

Constructors

new(embed_dim : Int32, num_heads : Int32, mlp_ratio : Float32 = 4.0_f32, with_cross_attn : Bool = false, dropout : Float32 = 0.0_f32, device : Tensor::Device = Tensor::Device::GPU)
Source

Instance methods

cross_attn
Source
forward(x : Autograd::Variable, other : Autograd::Variable | Nil = nil) : Autograd::Variable

Forward with optional cross-attention to another view

Source
mlp
Source
norm1
Source
norm2
Source
norm3
Source
parameters
Source
self_attn
Source