BackendBCEWithLogistLoss.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 CBackendBCEWithLogistLoss : public CBackendBase<T>
        {
        public:
            CBackendBCEWithLogistLoss();
            CBackendBCEWithLogistLoss(const CBackendBCEWithLogistLoss<T>& bl);
            virtual ~CBackendBCEWithLogistLoss();
 
            virtual const CResult BCEWithLogistLoss(const CTensor<T>* pTsrX, const CTensor<T>* pTsrTarget, CTensor<T>* pTsrResult, const CTensor<T>* pTsrWeightOrNull = nullptr, EReductionType eReductionType = EReductionType_Mean, const CTensor<T>* pTsrPosWeightOrNull = nullptr);
            virtual const CResult BCEWithLogistLoss_Backward(const CTensor<T>* pTsrX, const CTensor<T>* pTsrTarget, const CTensor<T>* pTsrDy, CTensor<T>* pTsrDx, const CTensor<T>* pTsrWeightOrNull = nullptr, EReductionType eReductionType = EReductionType_Mean, const CTensor<T>* pTsrPosWeightOrNull = nullptr);
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendBCEWithLogistLoss<T>, *this);
        };
    }
}
 
#endif