FLImaging 6.5.16.1
ComputationalGraphNormalRandom.h
1#pragma once
2
3#if _MSC_VER >= 1900 && defined(_M_X64)
4
5#include "ComputationalGraph.h"
6
7namespace FLImaging
8{
9 namespace AI
10 {
11 template <typename T>
12 class FL_EXPORT CComputationalGraphNormalRandom : public CComputationalGraph<T>
13 {
14 private:
15 CComputationalGraphNormalRandom();
16
17 protected:
18 CComputationalGraphNormalRandom(const CComputationalGraphNormalRandom<T>& cg);
19
20 public:
21
22 CComputationalGraphNormalRandom(const CComputationalBase<T>& cbOperand1, double f64Mean = 0., double f64Stdev = 1);
23 CComputationalGraphNormalRandom(const CTensor<T>& tsrOperand1, double f64Mean = 0., double f64Stdev = 1);
24 virtual ~CComputationalGraphNormalRandom();
25
26 virtual CTensor<T>& Forward() override;
27 virtual CTensor<T>* Backward() override;
28 virtual CComputationalBase<T>* Clone() const override;
29 virtual const CResult PrintNodeParamInfo() const override;
30
31 virtual const CResult GetBinaryData(Base::CFLData& fldBinary, bool bSuperClass = false, int32_t i32Version = -1, bool bDumpMode = false) const override;
32 virtual const CResult GetBinaryData(Base::CFLData* pFldBinary, bool bSuperClass = false, int32_t i32Version = -1, bool bDumpMode = false) const override;
33
34 virtual const CResult SetBinaryData(const Base::CFLData& fldBinary, int64_t* pI64Offset = nullptr) override;
35 virtual const CResult SetBinaryData(const Base::CFLData* pFldBinary, int64_t* pI64Offset = nullptr) override;
36
37 virtual const std::vector<int64_t>& GetEstimatedShape(bool bRecursive = true) const override;
38
39 virtual const CResult SetDistribution(double f64Mean = 0., double f64Stdev = 1.);
40 virtual const CResult GetDistribution(double& f64Mean, double& f64Stdev) const;
41
42
43 DeclareGetClassType();
44 SupportToDuplicateObjectWithoutCreateNewObject(CComputationalGraphNormalRandom, *this);
45
46 protected:
47 double m_f64Mean;
48 double m_f64Stdev;
49 public:
50 DeclareGetSignletonObject(CComputationalGraphNormalRandom);
51 };
52
53 #define CCGFNormalRandom(...) (*(new CComputationalGraphNormalRandom<float>(__VA_ARGS__)))
54 #define CCGDNormalRandom(...) (*(new CComputationalGraphNormalRandom<double>(__VA_ARGS__)))
55
56 #define CCGTNoramlRandom(T, ...) (*(new CComputationalGraphNormalRandom<T>(__VA_ARGS__)))
57 }
58}
59
60#endif