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, std::vector<int32_t> init_shape, std::vector<int32_t> init_strides, size_t init_offset = 0)
void *data() const

TODO(Documentation)

size_t bytes_count() const

TODO(Documentation)

std::vector<uint8_t> get_host_data() const

TODO(Documentation)

auto get_stream() const

TODO(Documentation)

Public Members

std::vector<int32_t> shape
std::vector<int32_t> strides
std::string typestr

Public Static Functions

static TensorObject create(std::shared_ptr<rmm::device_buffer> buffer, DType dtype, std::vector<TensorIndex> shape, std::vector<TensorIndex> strides, size_t offset = 0)

TODO(Documentation)

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.