#pragma once

#include "tensor.hpp"

namespace tf {

template <typename T>

class TensorNode {

};

template <typename T>

TensorNode<T>::Input::Input(Tensor<T>& in) :

tensor { std::shared_ptr<Tensor<T>>(&in, [](Tensor<T>*){}) } {

//std::cout << "input " << in.index() << '\n';

}

template <typename T>

TensorNode<T>::Output::Output(Tensor<T>& out) :

tensor { std::shared_ptr<Tensor<T>>(&out, [](Tensor<T>*){}) } {

//std::cout << "output " << out.index() << '\n';

}

template <typename T>

TensorNode<T>::Add::Add(TensorNode* l, TensorNode* r) :

lhs {l}, rhs {r} {

std::cout << "add: " << l << ' ' << r << '\n';

}

template <typename T>

template <typename... Args>

TensorNode<T>::TensorNode(Args&&... args) : _handle{std::forward<Args>(args)...} {

}

template <typename T>

void TensorNode<T>::_precede(TensorNode* v) {

_successors.push_back(v);

v->_dependents.push_back(this);

}

} // end of namespace tf -----------------------------------------------------