Class PANDecoderImpl

src/architectures/PANDecoder.h:44–55

}; TORCH_MODULE(GAUBlock);

class PANDecoderImpl:public torch::nn::Module
{
public:
	PANDecoderImpl(std::vector<int> encoder_channels, int decoder_channels, std::string upscale_mode = "bilinear");
	torch::Tensor forward(std::vector<torch::Tensor> x);
private:
	FPABlock fpa{ nullptr };
	GAUBlock gau3{ nullptr };
	GAUBlock gau2{ nullptr };
	GAUBlock gau1{ nullptr };

}; TORCH_MODULE(PANDecoder);