pytorch-inference
include/utils.hpp File Reference
#include <Python.h>
#include <numpy/arrayobject.h>
#include <stdexcept>
#include <arrayfire.h>

Go to the source code of this file.

Namespaces

 pytorch::internal
 

Macros

#define NPY_NO_DEPRECATED_API   NPY_1_7_API_VERSION
 

Functions

af::array pytorch::internal::from_numpy (PyArrayObject *array, int ndim, std::vector< int > dims)
 Converts a numpy array to an ArrayFire array. It is necessary to specify all 4 dimensions if there is a specific ordering required, otherwise it will be padded from the end with ones. More...
 
void pytorch::internal::check_size (const int &size1, const int &size2, const std::string &func)
 
void pytorch::internal::check_num_leq (const int &size1, const int &size2, const std::string &func)
 

Macro Definition Documentation

◆ NPY_NO_DEPRECATED_API

#define NPY_NO_DEPRECATED_API   NPY_1_7_API_VERSION