Class Tensor

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)

Previous Class SourceStageStopAfter
Next Class TensorMemory
© Copyright 2024, NVIDIA. Last updated on Apr 25, 2024.