NDarray Struct Reference

#include <JDFTxWrapper.h>

Public Member Functions
double &	operator[] (const std::vector< size_t > &index)

Public Attributes
double *	data
std::vector< size_t >	shape
std::vector< size_t >	strides
bool	onGpu

Detailed Description

Data structure to receive N-D buffers from Python (e.g. numpy.ndarray or torch.Tensor) Keeping this double-precision real only for now; can template this later if needed.

---

The documentation for this struct was generated from the following file:

• electronic/JDFTxWrapper.h