#pragma once

#include "tensor_expr.hpp"

#include "tensor_ops.hpp"

namespace tf {

template <typename T>

class TensorFrame {

};

template <typename T>

TensorExpr<T> TensorFrame<T>::input(Tensor<T>& tensor) {

return TensorExpr<T>(_tensor_nodes.emplace_back(std::make_unique<TensorNode<T>>(

std::in_place_type_t<typename TensorNode<T>::Input>{}, tensor

)).get());

}

template <typename T>

TensorExpr<T> TensorFrame<T>::output(Tensor<T>& tensor) {

return TensorExpr<T>(_tensor_nodes.emplace_back(std::make_unique<TensorNode<T>>(

std::in_place_type_t<typename TensorNode<T>::Output>{}, tensor

)).get());

}

template <typename T>

TensorExpr<T> TensorFrame<T>::add(TensorExpr<T> lhs, TensorExpr<T> rhs) {

auto res = TensorExpr<T>(_tensor_nodes.emplace_back(std::make_unique<TensorNode<T>>(

std::in_place_type_t<typename TensorNode<T>::Add>{},

lhs._tensor_node,

rhs._tensor_node

)).get());

res.succeed(lhs, rhs);

return res;

}

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