torch_normal.h

Go to the documentation of this file.

//
// Created by alex on 7/26/25.
//
 
#ifndef TORCH_NORMAL_H
#define TORCH_NORMAL_H
 
#include "cuberl/base/cubeai_config.h"
 
#ifdef USE_PYTORCH
 
#include "cuberl/base/cuberl_types.h"
#include "cuberl/maths/statistics/distributions/torch_distribution.h"
 
#include <torch/torch.h>
 
namespace cuberl
{
    namespace maths::stats
    {
        class TorchNormalDist final : public TorchDistributionBase
        {
 
        public:
 
            explicit TorchNormalDist(torch_tensor_t mu, torch_tensor_t sigma);
 
            torch_tensor_t sample()const;
 
 
            ~TorchNormalDist() override = default;
 
            torch_tensor_t log_prob(torch_tensor_t value) override;
 
            torch_tensor_t entropy() const override;
 
            torch_tensor_t mean()const {return mean_;}
 
            torch_tensor_t std()const {return sd_;}
        private:
 
            torch_tensor_t mean_;
            torch_tensor_t sd_;
 
            virtual torch_tensor_t sample(c10::ArrayRef<int64_t> /*sample_shape*/) { return torch_tensor_t(); }
        };
    }
}
 
#endif
 
#endif //TORCH_NORMAL_H