example_conv2dlayer_export.cpp

 
#include <Pds/Ra>
#include <Pds/Nn>
 
int main(void)
{
    unsigned int Nout=2;
    unsigned int Nin=2;
 
    // Tensor with Nout filters of Nin channels of size 3x3.
    Pds::TensorBlock W=Pds::Matrix::RandNTensorBlock(Nout,Nin,Pds::Size(3,3));
    
    // Creating a convolutional 2D Layer
    Pds::Conv2DLayer Conv(W,Pds::Identity,Pds::DIdentity,{1, 1},Pds::Ra::Same);
    
    // Saving the convolutional layer.
    std::ofstream myfile("Conv2DLayer.txt", std::ofstream::out);
    if (myfile.is_open())
    {
        Conv.SaveInStream(myfile);
        myfile.close();
    }
    
    // Loading the convolutional layer.
    Pds::Conv2DLayer Conv2;
    std::ifstream ifs("Conv2DLayer.txt");
    if (ifs.is_open())
    {
        Pds::Nn::DefaultActFunc=Pds::Tanh;
        Pds::Nn::DefaultDActFunc=Pds::DTanh;
        Conv2.LoadFromStream(ifs);
        ifs.close();
    }
    
    Conv2.PrintStylized("Conv:\n");
    
    return 0;
}