class

GS::MASt3R::Reassemble

Inherits Reference < Object

Feature fusion module (reassemble)

Constructors

new(in_features : Int32, out_features : Int32, device : Tensor::Device = Tensor::Device::GPU)
Source

Instance methods

forward(x : Autograd::Variable, height : Int32, width : Int32, skip_cls : Bool = true) : Autograd::Variable

Reassemble tokens back to spatial features x: [batch, num_tokens, embed_dim] Returns: [batch, height, width, out_features]

Source
parameters
Source
proj
Source