JNIEXPORT jint JNICALL Java_ml_dmlc_mxnet_LibInfo_mxNDArrayCreateNone(JNIEnv *env, jobject obj, jobject ndArrayHandle) { NDArrayHandle out; int ret = MXNDArrayCreateNone(&out); jclass ndClass = env->GetObjectClass(ndArrayHandle); jfieldID ptr = env->GetFieldID(ndClass, "value", "J"); env->SetLongField(ndArrayHandle, ptr, (long)out); return ret; }
inline NDArray::NDArray(const mx_float *data, size_t size) { NDArrayHandle handle; CHECK_EQ(MXNDArrayCreateNone(&handle), 0); MXNDArraySyncCopyFromCPU(handle, data, size); blob_ptr_ = std::make_shared<NDBlob>(handle); }
inline NDArray::NDArray(const std::vector<mx_float> &data) { NDArrayHandle handle; CHECK_EQ(MXNDArrayCreateNone(&handle), 0); MXNDArraySyncCopyFromCPU(handle, data.data(), data.size()); blob_ptr_ = std::make_shared<NDBlob>(handle); }
inline NDArray::NDArray() { NDArrayHandle handle; CHECK_EQ(MXNDArrayCreateNone(&handle), 0); blob_ptr_ = std::make_shared<NDBlob>(handle); }