RaggedTensor Struct Reference

#include <RaggedTensor.h>

Inheritance diagram for RaggedTensor:

Public Member Functions
	RaggedTensor ()
	RaggedTensor (torch::Tensor values, torch::Tensor row_splits)
	Constructor for creating RaggedTensor with values and row_splits. More...
c10::intrusive_ptr< RaggedTensor >	FromRowSplits (torch::Tensor values, torch::Tensor row_splits, bool validate=true) const
torch::Tensor	GetValues () const
	Returns _values tensor. More...
torch::Tensor	GetRowSplits () const
	Returns _row_splits tensor. More...
std::string	ToString () const
	Returns string representation. More...
torch::Tensor	GetItem (int key) const
int64_t	Len () const
c10::intrusive_ptr< RaggedTensor >	Clone () const
	Copy Tensor to the same device. More...
c10::intrusive_ptr< RaggedTensor >	Concat (c10::intrusive_ptr< RaggedTensor > r_tensor, int64_t axis) const
template<typename T >
c10::intrusive_ptr< RaggedTensor >	Add (T value) const
template<typename T >
c10::intrusive_ptr< RaggedTensor >	Add_ (T value)
template<typename T >
c10::intrusive_ptr< RaggedTensor >	Sub (T value) const
template<typename T >
c10::intrusive_ptr< RaggedTensor >	Sub_ (T value)
template<typename T >
c10::intrusive_ptr< RaggedTensor >	Mul (T value) const
template<typename T >
c10::intrusive_ptr< RaggedTensor >	Mul_ (T value)
template<typename T >
c10::intrusive_ptr< RaggedTensor >	Div (T value) const
template<typename T >
c10::intrusive_ptr< RaggedTensor >	Div_ (T value)
template<typename T >
c10::intrusive_ptr< RaggedTensor >	FloorDiv (T value) const
template<typename T >
c10::intrusive_ptr< RaggedTensor >	FloorDiv_ (T value)

Detailed Description

A RaggedTensor is a tensor with ragged dimensions, whose slice may have different lengths. We define a container for ragged tensor to support operations involving batches whose elements may have different shape.

Constructor & Destructor Documentation

RaggedTensor() [1/2]

RaggedTensor::RaggedTensor ( )

inline

RaggedTensor() [2/2]

RaggedTensor::RaggedTensor ( torch::Tensor  values, torch::Tensor  row_splits  )

inline

Constructor for creating RaggedTensor with values and row_splits.

Member Function Documentation

Add()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::Add ( T  value ) const

inline

Add_()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::Add_ ( T  value )

inline

Clone()

c10::intrusive_ptr< RaggedTensor > RaggedTensor::Clone

(

)

const

Copy Tensor to the same device.

Concat()

c10::intrusive_ptr< RaggedTensor > RaggedTensor::Concat	(	c10::intrusive_ptr< RaggedTensor >	r_tensor,
		int64_t	axis
	)		const

Div()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::Div ( T  value ) const

inline

Div_()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::Div_ ( T  value )

inline

FloorDiv()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::FloorDiv ( T  value ) const

inline

FloorDiv_()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::FloorDiv_ ( T  value )

inline

FromRowSplits()

c10::intrusive_ptr< RaggedTensor > RaggedTensor::FromRowSplits	(	torch::Tensor	values,
		torch::Tensor	row_splits,
		bool	validate = true
	)		const

Creates a RaggedTensor with rows partitioned by row_splits.

The returned RaggedTensor corresponds with the python list defined by:

result = [values[row_splits[i]:row_splits[i + 1]]
          for i in range(len(row_splits) - 1)]

GetItem()

torch::Tensor RaggedTensor::GetItem

(

int

key

)

const

Pythonic getitem for RaggedTensor.

Returns a slice of values based on row_splits. It can be used to retrieve i'th batch element. Currently it only supports a single integer index.

GetRowSplits()

torch::Tensor RaggedTensor::GetRowSplits

(

)

const

Returns _row_splits tensor.

GetValues()

torch::Tensor RaggedTensor::GetValues

(

)

const

Returns _values tensor.

Len()

int64_t RaggedTensor::Len

(

)

const

Pythonic len for RaggedTensor.

Returns number of batch elements.

Mul()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::Mul ( T  value ) const

inline

Mul_()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::Mul_ ( T  value )

inline

Sub()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::Sub ( T  value ) const

inline

Sub_()

template<typename T >

c10::intrusive_ptr<RaggedTensor> RaggedTensor::Sub_ ( T  value )

inline

ToString()

std::string RaggedTensor::ToString

(

)

const

Returns string representation.

---

The documentation for this struct was generated from the following files:

• /home/runner/work/Open3D/Open3D/cpp/open3d/ml/pytorch/ragged_tensor/RaggedTensor.h (b9e049c (Sun Dec 31 11:36:26 2023 -0800))
• /home/runner/work/Open3D/Open3D/cpp/open3d/ml/pytorch/ragged_tensor/RaggedTensor.cpp (b9e049c (Sun Dec 31 11:36:26 2023 -0800))