Class Tensor

• Defined in File tensor.hpp

Class Documentation

class Tensor

Morpheus Tensor object, using RMM device buffers for the underlying storage. Typically created using the Tensor::create factory method and accessed through a TensorObject handle.

Public Functions

Tensor(

std::shared_ptr<rmm::device_buffer> buffer,

std::string init_typestr,

ShapeType init_shape,

ShapeType init_strides,

TensorSize init_offset = 0

)

void *data() const

TODO(Documentation)

TensorSize bytes_count() const

TODO(Documentation)

std::vector<uint8_t> get_host_data() const

TODO(Documentation)

auto get_stream() const

TODO(Documentation)

Public Members

ShapeType shape

ShapeType strides

std::string typestr

Public Static Functions

static TensorObject create(

std::shared_ptr<rmm::device_buffer> buffer,

DType dtype,

ShapeType shape,

ShapeType strides,

TensorSize offset = 0

)

TODO(Documentation)