org.bytedeco.tvm

Class NDArray

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tvm.ObjectRef

org.bytedeco.tvm.NDArray

All Implemented Interfaces:

AutoCloseable

@Namespace(value="tvm::runtime")
 @Properties(inherit=tvm_runtime.class)
public class NDArray
extends ObjectRef

\brief Managed NDArray. The array is backed by reference counted blocks.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	NDArray.Container \brief NDArray internal container type
static class	NDArray.ContainerBase \brief ContainerBase used to back the TVMArrayHandle
static class	NDArray.Internal

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.tvm.ObjectRef

_type_is_nullable

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
NDArray() \brief default constructor
NDArray(long size) Native array allocator.
NDArray(ObjectPtr data) \brief constructor.
NDArray(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
DLTensor	accessDLTensor()
void	CopyFrom(DLTensor other) \brief Copy data content from another array.
void	CopyFrom(NDArray other)
void	CopyFromBytes(Pointer data, long nbytes) \brief Copy data content from a byte buffer.
static void	CopyFromTo(DLTensor from, DLTensor to)
static void	CopyFromTo(DLTensor from, DLTensor to, TVMStreamHandle stream) \brief Function to copy data from one array to another.
NDArray	CopyTo(DLContext ctx) \brief Copy the data to another context.
void	CopyTo(DLTensor other) \brief Copy data content into another array.
void	CopyTo(NDArray other)
void	CopyToBytes(Pointer data, long nbytes) \brief Copy data content into another array.
NDArray	CreateView(long[] shape, DLDataType dtype)
NDArray	CreateView(LongBuffer shape, DLDataType dtype)
NDArray	CreateView(LongPointer shape, DLDataType dtype) \brief Create a NDArray that shares the data memory with the current one.
DataType	DataType()
static NDArray	Empty(long[] shape, DLDataType dtype, DLContext ctx)
static NDArray	Empty(LongBuffer shape, DLDataType dtype, DLContext ctx)
static NDArray	Empty(LongPointer shape, DLDataType dtype, DLContext ctx) \brief Create an empty NDArray.
static NDArray	FromDLPack(DLManagedTensor tensor) \brief Create a NDArray backed by a dlpack tensor.
NDArray	getPointer(long i)
boolean	IsContiguous()
boolean	Load(Stream stream) \brief Load NDArray from stream
NDArray	position(long position)
void	Save(Stream stream) \brief Save NDArray to stream
LongPointer	Shape()
DLManagedTensor	ToDLPack() \brief Create a reference view of NDArray that represents as DLManagedTensor.
int	use_count()

Methods inherited from class org.bytedeco.tvm.ObjectRef

_type_is_nullable, access, defined, equals, get, lessThan, notEquals, same_as, unique

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getPointer, getPointer, getPointer, hashCode, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

NDArray

public NDArray(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

NDArray

public NDArray(long size)

Native array allocator. Access with Pointer.position(long).

NDArray

public NDArray()

\brief default constructor

NDArray

public NDArray(@ByVal
               ObjectPtr data)

\brief constructor.

Parameters:

data - ObjectPtr to the data container.

Method Detail

position

public NDArray position(long position)

Overrides:

position in class ObjectRef

getPointer

public NDArray getPointer(long i)

Overrides:

getPointer in class ObjectRef

use_count

public int use_count()

Overrides:

use_count in class ObjectRef

Returns:

the reference counter \note this number is approximate in multi-threaded setting.

accessDLTensor

@Const
 @Name(value="operator ->")
public DLTensor accessDLTensor()

Returns:

Pointer to content of DLTensor

IsContiguous

@Cast(value="bool")
public boolean IsContiguous()

Returns:

Whether the tensor is contiguous

CopyFrom

public void CopyFrom(@Const
                     DLTensor other)

\brief Copy data content from another array.

Parameters:

other - The source array to be copied from. \note The copy may happen asynchronously if it involves a GPU context. TVMSynchronize is necessary.

CopyFrom

public void CopyFrom(@Const @ByRef
                     NDArray other)

CopyFromBytes

public void CopyFromBytes(@Const
                          Pointer data,
                          @Cast(value="size_t")
                          long nbytes)

\brief Copy data content from a byte buffer.

Parameters:

data - The source bytes to be copied from.

nbytes - The size of the buffer in bytes Must be equal to the size of the NDArray. \note The copy always triggers a TVMSynchronize.

CopyTo

public void CopyTo(DLTensor other)

\brief Copy data content into another array.

Parameters:

other - The source array to be copied from. \note The copy may happen asynchronously if it involves a GPU context. TVMSynchronize is necessary.

CopyTo

public void CopyTo(@Const @ByRef
                   NDArray other)

CopyToBytes

public void CopyToBytes(Pointer data,
                        @Cast(value="size_t")
                        long nbytes)

\brief Copy data content into another array.

Parameters:

data - The source bytes to be copied from.

nbytes - The size of the data buffer. Must be equal to the size of the NDArray. \note The copy always triggers a TVMSynchronize.

CopyTo

@ByVal
public NDArray CopyTo(@Const @ByRef
                              DLContext ctx)

\brief Copy the data to another context.

Parameters:

ctx - The target context.

Returns:

The array under another context.

Load

@Cast(value="bool")
public boolean Load(Stream stream)

\brief Load NDArray from stream

Parameters:

stream - The input data stream

Returns:

Whether load is successful

Save

public void Save(Stream stream)

\brief Save NDArray to stream

Parameters:

stream - The output data stream

CreateView

@ByVal
public NDArray CreateView(@Cast(value="int64_t*") @StdVector
                                  LongPointer shape,
                                  @ByVal
                                  DLDataType dtype)

\brief Create a NDArray that shares the data memory with the current one.

Parameters:

shape - The shape of the new array.

dtype - The data type of the new array. \note The memory size of new array must be smaller than the current one.

CreateView

@ByVal
public NDArray CreateView(@Cast(value="int64_t*") @StdVector
                                  LongBuffer shape,
                                  @ByVal
                                  DLDataType dtype)

CreateView

@ByVal
public NDArray CreateView(@Cast(value="int64_t*") @StdVector
                                  long[] shape,
                                  @ByVal
                                  DLDataType dtype)

ToDLPack

public DLManagedTensor ToDLPack()

\brief Create a reference view of NDArray that represents as DLManagedTensor.

Returns:

A DLManagedTensor

Empty

@ByVal
public static NDArray Empty(@Cast(value="int64_t*") @StdVector
                                    LongPointer shape,
                                    @ByVal
                                    DLDataType dtype,
                                    @ByVal
                                    DLContext ctx)

\brief Create an empty NDArray.

Parameters:

shape - The shape of the new array.

dtype - The data type of the new array.

ctx - The context of the Array.

Returns:

The created Array

Empty

@ByVal
public static NDArray Empty(@Cast(value="int64_t*") @StdVector
                                    LongBuffer shape,
                                    @ByVal
                                    DLDataType dtype,
                                    @ByVal
                                    DLContext ctx)

Empty

@ByVal
public static NDArray Empty(@Cast(value="int64_t*") @StdVector
                                    long[] shape,
                                    @ByVal
                                    DLDataType dtype,
                                    @ByVal
                                    DLContext ctx)

FromDLPack

@ByVal
public static NDArray FromDLPack(DLManagedTensor tensor)

\brief Create a NDArray backed by a dlpack tensor. This allows us to create a NDArray using the memory allocated by an external deep learning framework that is DLPack compatible. The memory is retained until the NDArray went out of scope.

Parameters:

tensor - The DLPack tensor to copy from.

Returns:

The created NDArray view.

CopyFromTo

public static void CopyFromTo(@Const
                              DLTensor from,
                              DLTensor to,
                              TVMStreamHandle stream)

\brief Function to copy data from one array to another.

Parameters:

from - The source array.

to - The target array.

stream - The stream used in copy.

CopyFromTo

public static void CopyFromTo(@Const
                              DLTensor from,
                              DLTensor to)

Shape

@Cast(value="int64_t*")
 @StdVector
public LongPointer Shape()

DataType

@ByVal
public DataType DataType()