coral.py

class CORALLossLayer(object):
    def __init__(self, diff_, color, cov_s, cov_t, mean_s, mean_t, diff_data_s, diff_data_t, square_mean_s, square_mean_t, bp_mean_s, bp_mean_t, bp_der_s, bp_der_t, identity):
        
        self.diff_=diff_;
        self.cov_s=cov_s;
        self.cov_t=cov_t;
        self.mean_s=mean_s;
        self.mean_t=mean_t;
        self.diff_data_s=diff_data_s;
        self.diff_data_t=diff_data_t;
        self.square_mean_s=square_mean_s;
        self.square_mean_t=square_mean_t;
        self.bp_mean_s=bp_mean_s;
        self.bp_mean_t=bp_mean_t;
        self.bp_der_s=bp_der_s;
        self.bp_der_t=bp_der_t;
        self.identity=identity;

    def Reshape(bottom,top):
        return;
    

    def type() : return "CORALLoss"; 
  
    def AllowForceBackward(bottom_index):
            return true;
    def Forward_cpu(bottom,top):
        return;
    def Backward_cpu(top,propagate_down,bottom):
        return;