variable.h

Go to the documentation of this file.

#pragma once
#include <string>
#include <tensor/tensor.h>
#include "operation.h"

namespace magmadnn {
namespace op {

template <typename T>
class Variable : public Operation<T> {
public:
    Variable (std::string name, std::vector<unsigned int> shape, tensor_filler_t<T> filler, memory_t mem_type);
    Variable (std::string name, Tensor<T> *val);
    ~Variable();

    std::string to_string() { return name; }
    std::string get_name() { return name; }

protected:
    Tensor<T> *_eval(bool recompute=true);
    Tensor<T> *_grad(Operation<T> *consumer, Operation<T> *var, Tensor<T> *grad);

    std::string name;
    Tensor<T> *val;
    bool delete_tensor;

};

template <typename T>
Variable<T>* var(std::string name, Tensor<T>* val);

template <typename T>
Variable<T>* var(std::string name, std::vector<unsigned int> shape, tensor_filler_t<T> filler, memory_t mem_type);

template <typename T>
Variable<T> *scalar(std::string name, T val, memory_t mem_type);

} // namespace op
} // namespace magmadnn