Function getDLTensor

python/astra/src/dlpack.cpp:248–261

}

astra::CFloat32VolumeData2D* getDLTensor(PyObject *obj, const astra::CVolumeGeometry2D &pGeom, std::string &error)
{
	if (!PyCapsule_CheckExact(obj))
		return nullptr;

	// x,y,z
	std::array<int, 2> dims{pGeom.getGridColCount(), pGeom.getGridRowCount()};

	astra::CDataStorage *storage = getDLTensorStorage(obj, dims, error);
	if (!storage)
		return nullptr;

	return new astra::CFloat32VolumeData2D(pGeom, storage);
}
astra::CFloat32ProjectionData2D* getDLTensor(PyObject *obj, const astra::CProjectionGeometry2D &pGeom, std::string &error)
{
	if (!PyCapsule_CheckExact(obj))