BackendCTCLoss.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "BackendBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CTensor;
 
        template <typename T>
        class FL_EXPORT CBackendCTCLoss : public CBackendBase<T>
        {
        public:
            CBackendCTCLoss();
            CBackendCTCLoss(const CBackendCTCLoss<T>& bl);
            virtual ~CBackendCTCLoss();
 
            virtual const CResult SetReductionType(EReductionType eReductionType);
            virtual EReductionType GetReductionType() const;
 
            virtual const CResult SetBlankLabel(int64_t i64BlankLabel);
            virtual int64_t GetBlankLabel() const;
 
            virtual const CResult EnableZeroInfinity(bool bZeroInfinity);
            virtual bool IsZeroInfinityEnable() const;
 
            virtual const CResult Forward(const CTensor<T>* pTsrX, const CTensor<T>* pTsrTarget, CTensor<T>* pTsrProbablity, CTensor<T>* pTsrBatchLoss, CTensor<T>* pTsrResult);
            virtual const CResult Backward(const CTensor<T>* pTsrDy, const CTensor<T>* pTsrX, const CTensor<T>* pTsrTarget, const CTensor<T>* pTsrForwardProbablity, const CTensor<T>* pTsrBatchLoss, CTensor<T>* pTsrDx, bool bAddGradient);
            
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendCTCLoss<T>, *this);
 
        protected:
 
            inline T GetLogInnerAdd(T tValue1, T tValue2) const;
            inline T GetLogInnerAdd_DynamicType(T tValue1, T tValue2) const;
            
            EReductionType m_eReductionType;
            int64_t m_i64BlankLabel;
            bool m_bZeroInfinity;
        };
    }
}
 
#endif