example_flattenlayer_predict.cpp

 
#include <Pds/Ra>
#include <Pds/Nn>
 
int main(void)
{
    Pds::FlattenLayer Layer;
    Pds::Matrix M0=Pds::RandN(1,2);
    Pds::Matrix M1=Pds::RandN(1,2);
    Pds::SampleBlock Sample={M0,M1};
    Sample[0].Print("Sample[0]:\n");
    Sample[1].Print("Sample[1]:\n");
    
    std::cout<<"\n";
    
    Pds::Vector Out=Layer.Predict(Sample);
    Out.T().Print("Out.T():\n");
    
    std::cout<<"\n";
    
    Pds::TensorBlock Tensor=Layer.PredictTensor({Sample});
    Tensor[0][0].Print("Tensor[0][0]:\n");
    
    return 0;
}