canvas-query-ranker.h File Reference

#include <torch/script.h>
#include <torch/torch.h>
#include <cereal/types/complex.hpp>
#include <cereal/types/string.hpp>
#include <cereal/types/vector.hpp>
#include <algorithm>
#include <cmath>
#include <cstdint>
#include <iostream>
#include <memory>
#include <string>
#include <vector>
#include "common.h"
#include "image-processor.h"
#include "keyword-ranker.h"
#include "query-types.h"

Include dependency graph for canvas-query-ranker.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	sh::CanvasQueryRanker

Namespaces
	sh

Functions
template<typename DType >
std::vector< std::vector< DType > >	sh::to_std_matrix (const at::Tensor &tensor_features)
template<c10::ScalarType TensorDType_ = at::kFloat, typename OrigDType_ = float>
at::Tensor	sh::to_tensor (std::vector< OrigDType_ > &orig_vec)
template<c10::ScalarType TensorDType_ = at::kFloat, typename OrigDType_ = float>
at::Tensor	sh::to_tensor (std::vector< std::vector< OrigDType_ >> &orig_mat)