class

GS::MASt3R::Conv2DBlock

Inherits Reference < Object

Conv2D block for DPT

Constructors

new(in_channels : Int32, out_channels : Int32, kernel_size : Int32 = 3, stride : Int32 = 1, padding : Int32 = 1, bias : Bool = true, device : Tensor::Device = Tensor::Device::GPU)
Source

Instance methods

bias
Source
forward(x : Autograd::Variable) : Autograd::Variable

Convolution forward (GPU when available, CPU fallback) x: [batch, height, width, in_channels]

Source
in_channels
Source
kernel_size
Source
out_channels
Source
padding
Source
parameters
Source
stride
Source
weight
Source