rlmesh::Tensor
Class in rlmesh.hpp.
class TensorConstructors
Static methods
Methods
- raw()
const RLMeshTensor* raw() const - dtype()
DType dtype() const - device()
Device device() const - shape()
std::vector<size_t> shape() const - strides()
std::vector<size_t> strides() const - ndim()
size_t ndim() const - size()
size_t size() const - nbytes()
size_t nbytes() const - buffer()
Span<const std::byte> buffer() const - is_contiguous()
bool is_contiguous() const - reshape()
Tensor reshape(std::vector<int64_t> shape) const - copy()
Tensor copy() const - tobytes()
std::vector<std::byte> tobytes() const - to_string()
std::string to_string() const - to_dlpack()
ext::DlpackPtr to_dlpack(ToDlpackOptions options = {}) const - dlpack_device()
std::pair<int32_t, int32_t> dlpack_device() const
Tensor
constructor[source]explicit Tensor(RLMeshTensor* p)Tensor
constructor[source]Tensor(
const void* buffer,
size_t buffer_len,
std::vector<size_t> shape,
DType dtype
)from_dlpack
static method[source]static Tensor from_dlpack(ext::DlpackPtr&& obj)raw
method[source]const RLMeshTensor* raw() constdtype
method[source]DType dtype() constdevice
method[source]Device device() constshape
method[source]std::vector<size_t> shape() conststrides
method[source]std::vector<size_t> strides() constndim
method[source]size_t ndim() constsize
method[source]size_t size() constnbytes
method[source]size_t nbytes() constbuffer
method[source]Span<const std::byte> buffer() constis_contiguous
method[source]bool is_contiguous() constreshape
method[source]Tensor reshape(std::vector<int64_t> shape) constcopy
method[source]Tensor copy() consttobytes
method[source]std::vector<std::byte> tobytes() constto_string
method[source]std::string to_string() constto_dlpack
method[source]ext::DlpackPtr to_dlpack(ToDlpackOptions options = {}) constdlpack_device
method[source]std::pair<int32_t, int32_t> dlpack_device() const