example_conv2dlayer_predict2.cpp

 
#include <Pds/Ra>
#include <Pds/Nn>
 
int main(void)
{
    Pds::SampleBlock Out;
    unsigned int L=5;
    
    // Input file
    Pds::SampleBlock In={Pds::RandN(5,5),Pds::RandN(5,5),Pds::RandN(5,5)};
    In[0].Print("In[0]:\n");
    In[1].Print("In[1]:\n");
    In[2].Print("In[2]:\n");
    
    
    std::cout<<"\n";
    
    // Filtro Mean
    Pds::Matrix Mean=Pds::Kernel2D::Mean(L)/3;
    
    // Tensor con 2 filtros de 3 capas.
    Pds::TensorBlock W={{Mean,Mean,Mean}};
    
    Pds::Conv2DLayer Conv(W,Pds::Identity,Pds::DIdentity,{1, 1},Pds::Ra::Same);
    
    Out=Conv.Predict(In);
    Out[0].Print("Out[0]:\n");
    
    Pds::TensorBlock Block=Conv.PredictTensor({In});
    Block[0][0].Print("Block[0][0]:\n");
    return 0;
}