mxnet
|
Tensor data structure used by custom operator. More...
#include <lib_api.h>
Public Member Functions | |
MXTensor () | |
MXTensor (const MXTensor &oth) | |
MXTensor (void *data_ptr, const std::vector< int64_t > &shape, MXDType dtype, size_t vID, MXContext mx_ctx, MXStorageType stype=kDefaultStorage) | |
void | setTensor (void *dptr, MXDType type, const int64_t *dims, int ndims, size_t vID, MXContext mx_ctx, MXStorageType storage_type) |
populate internal tensor fields More... | |
void | setDLTensor () |
populate DLTensor fields More... | |
template<typename data_type > | |
data_type * | data () |
helper function to cast data pointer More... | |
int64_t | size () const |
helper function to get data size More... | |
bool | isSame (const MXTensor &oth) const |
helper function to compare two MXTensors More... | |
Public Attributes | |
void * | data_ptr |
std::vector< int64_t > | shape |
MXDType | dtype |
size_t | verID |
MXContext | ctx |
DLTensor | dltensor |
MXStorageType | stype |
Tensor data structure used by custom operator.
|
inline |
|
inline |
|
inline |
|
inline |
helper function to cast data pointer
|
inline |
helper function to compare two MXTensors
|
inline |
populate DLTensor fields
|
inline |
populate internal tensor fields
|
inline |
helper function to get data size
MXContext MXTensor::ctx |
void* MXTensor::data_ptr |
DLTensor MXTensor::dltensor |
MXDType MXTensor::dtype |
std::vector<int64_t> MXTensor::shape |
MXStorageType MXTensor::stype |
size_t MXTensor::verID |