Class LargeVis

include/instance/model/visualization.h:34–86

 */
template<class _Vector>
class LargeVis {
public:
    static const size_t dim = _Vector::dim;
    typedef _Vector Vector;
    typedef typename _Vector::Float Float;

    __host__ __device__
    static void forward(const Vector &head, const Vector &tail, Float &output) {
        output = 0;
        FOR(i, dim)
            output += (head[i] - tail[i]) * (head[i] - tail[i]);
        output = SUM(output);
    }

    template<OptimizerType optimizer_type>
    __host__ __device__
    static void backward(Vector &head, Vector &tail, Float gradient, const Optimizer &optimizer, Float weight = 1) {
        auto update = get_update_function<Float, optimizer_type>();
        FOR(i, dim) {
            Float h = head[i];
            Float t = tail[i];
            head[i] -= (optimizer.*update)(h, gradient * (h - t), weight);
            tail[i] -= (optimizer.*update)(t, gradient * (t - h), weight);
        }
    }

    template<OptimizerType optimizer_type>
    __host__ __device__
    static void backward(Vector &head, Vector &tail, Vector &head_moment1, Vector &tail_moment1,
                         Float gradient, const Optimizer &optimizer, Float weight = 1) {
        auto update = get_update_function_1_moment<Float, optimizer_type>();
        FOR(i, dim) {
            Float h = head[i];
            Float t = tail[i];
            head[i] -= (optimizer.*update)(h, gradient * (h - t), head_moment1[i], weight);
            tail[i] -= (optimizer.*update)(t, gradient * (t - h), tail_moment1[i], weight);
        }
    }

    template<OptimizerType optimizer_type>
    __host__ __device__
    static void backward(Vector &head, Vector &tail, Vector &head_moment1, Vector &tail_moment1,
                         Vector &head_moment2, Vector &tail_moment2,
                         Float gradient, const Optimizer &optimizer, Float weight = 1) {
        auto update = get_update_function_2_moment<Float, optimizer_type>();
        FOR(i, dim) {
            Float h = head[i];
            Float t = tail[i];
            head[i] -= (optimizer.*update)(h, gradient * (h - t), head_moment1[i], head_moment2[i], weight);
            tail[i] -= (optimizer.*update)(t, gradient * (t - h), tail_moment1[i], tail_moment2[i], weight);
        }
    }
};

}