BackendConcatenate.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "BackendBase.h"
#include "Tensor.h"
#include <vector>
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CComputationalBase;
 
        template <typename T>
        class FL_EXPORT CTensor;
 
        template <typename T>
        class FL_EXPORT CBackendConcatenate : public CBackendBase<T>
        {
        public:
            CBackendConcatenate();
            CBackendConcatenate(const CBackendConcatenate<T>& bl);
            virtual ~CBackendConcatenate();
 
 
            virtual const CResult Concatenate(const std::initializer_list<CTensor<T>>& vctTarget, CTensor<T>* pTsrResult, int64_t i64Axis);
            virtual const CResult Concatenate(const std::vector<CTensor<T>>& vctTarget, CTensor<T>* pTsrResult, int64_t i64Axis);
            virtual const CResult Concatenate(const std::initializer_list<CTensor<T>*>& ilTarget, CTensor<T>* pTsrResult, int64_t i64Axis);
            virtual const CResult Concatenate(const std::vector<CTensor<T>*>& vctTarget, CTensor<T>* pTsrResult, int64_t i64Axis);
            
            virtual const CResult ConcatenateBackward(const CTensor<T>* pTsrDy, const std::vector<bool> vctActivationFlag, std::vector<CComputationalBase<T>*>& vctCB, int64_t i64Axis);
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendConcatenate<T>, *this);
        };
    }
}
 
#endif