Open3D (C++ API)  0.17.0
Public Types | Public Member Functions | Protected Attributes
open3d::core::AdvancedIndexer Class Reference

#include <AdvancedIndexing.h>

Public Types

enum class  AdvancedIndexerMode { SET , GET }
 

Public Member Functions

 AdvancedIndexer (const Tensor &src, const Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides, AdvancedIndexerMode mode)
 
OPEN3D_HOST_DEVICE char * GetInputPtr (int64_t workload_idx) const
 
OPEN3D_HOST_DEVICE char * GetOutputPtr (int64_t workload_idx) const
 
OPEN3D_HOST_DEVICE int64_t GetIndexedOffset (int64_t workload_idx) const
 
int64_t NumWorkloads () const
 

Protected Attributes

Indexer indexer_
 
AdvancedIndexerMode mode_
 
int64_t num_indices_
 
int64_t element_byte_size_
 
int64_t indexed_shape_ [MAX_DIMS]
 
int64_t indexed_strides_ [MAX_DIMS]
 

Detailed Description

AdvancedIndexer exposes an interface similar to Indexer, with support for advanced indexing.

In particular, AdvancedIndexer contains an Indexer instance to iterate src, dst and index tensors. AdvancedIndexer also contains various properties for advanced indexing.

To construct AdvancedIndexer, one must use the tensors and index tensors preprocessed by AdvancedIndexPreprocessor.

Member Enumeration Documentation

◆ AdvancedIndexerMode

Enumerator
SET 
GET 

Constructor & Destructor Documentation

◆ AdvancedIndexer()

open3d::core::AdvancedIndexer::AdvancedIndexer ( const Tensor src,
const Tensor dst,
const std::vector< Tensor > &  index_tensors,
const SizeVector indexed_shape,
const SizeVector indexed_strides,
AdvancedIndexerMode  mode 
)
inline

Member Function Documentation

◆ GetIndexedOffset()

OPEN3D_HOST_DEVICE int64_t open3d::core::AdvancedIndexer::GetIndexedOffset ( int64_t  workload_idx) const
inline

◆ GetInputPtr()

OPEN3D_HOST_DEVICE char* open3d::core::AdvancedIndexer::GetInputPtr ( int64_t  workload_idx) const
inline

◆ GetOutputPtr()

OPEN3D_HOST_DEVICE char* open3d::core::AdvancedIndexer::GetOutputPtr ( int64_t  workload_idx) const
inline

◆ NumWorkloads()

int64_t open3d::core::AdvancedIndexer::NumWorkloads ( ) const
inline

Field Documentation

◆ element_byte_size_

int64_t open3d::core::AdvancedIndexer::element_byte_size_
protected

◆ indexed_shape_

int64_t open3d::core::AdvancedIndexer::indexed_shape_[MAX_DIMS]
protected

◆ indexed_strides_

int64_t open3d::core::AdvancedIndexer::indexed_strides_[MAX_DIMS]
protected

◆ indexer_

Indexer open3d::core::AdvancedIndexer::indexer_
protected

◆ mode_

AdvancedIndexerMode open3d::core::AdvancedIndexer::mode_
protected

◆ num_indices_

int64_t open3d::core::AdvancedIndexer::num_indices_
protected

The documentation for this class was generated from the following file: