wehub-resource-sync
35 lines
738 B
C++
35 lines
738 B
C++
//
|
|
// Created by agibsonccc on 4/26/22.
|
|
//
|
|
|
|
#ifndef LIBND4J_NDINDEXUTILS_H
|
|
#define LIBND4J_NDINDEXUTILS_H
|
|
|
|
#include <system/common.h>
|
|
#include <array/NDArrayFactory.h>
|
|
#include <array/NDArray.h>
|
|
|
|
|
|
namespace sd {
|
|
|
|
#define POINT_TYPE 0
|
|
#define INTERVAL_TYPE 1
|
|
#define ALL_TYPE 2
|
|
#define NEW_AXIS 3
|
|
|
|
#define DEFAULT_INCLUSIVE 1
|
|
|
|
class SD_LIB_EXPORT NDIndexUtils {
|
|
|
|
public:
|
|
static NDArray createInterval(LongType start, LongType end, LongType stride = 1, LongType inclusive = 1);
|
|
static NDArray createInterval(LongType start, LongType end, LongType stride = 1, bool inclusive = true);
|
|
static NDArray createPoint(LongType offset);
|
|
static NDArray createNewAxis();
|
|
static NDArray createAll();
|
|
|
|
};
|
|
}
|
|
|
|
#endif // LIBND4J_NDINDEXUTILS_H
|