class

GS::MASt3R::ConvBlock

Inherits Reference < Object

Convolution-like operation (implemented via unfold + matmul) For simplicity, uses pixel shuffle for upsampling

Constructors

new(in_channels : Int32, out_channels : Int32, with_norm : Bool = true, device : Tensor::Device = Tensor::Device::GPU)
Source

Instance methods

conv
Source
forward(x : Autograd::Variable) : Autograd::Variable
Source
norm
Source
parameters
Source