deepks_check.h File Reference

#include <string>
#include <torch/script.h>
#include <torch/torch.h>

Include dependency graph for deepks_check.h:

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

Go to the source code of this file.

Namespaces
namespace	DeePKS_domain

Functions
template<typename T >
void	DeePKS_domain::check_tensor (const torch::Tensor &tensor, const std::string &filename, const int rank)