Open3D (C++ API)  0.18.0+252c867
IndexGetSet.h
Go to the documentation of this file.
1 // ----------------------------------------------------------------------------
2 // - Open3D: www.open3d.org -
3 // ----------------------------------------------------------------------------
4 // Copyright (c) 2018-2023 www.open3d.org
5 // SPDX-License-Identifier: MIT
6 // ----------------------------------------------------------------------------
7 
8 #pragma once
9 
10 #include "open3d/core/Tensor.h"
11 #include "open3d/utility/Logging.h"
12 
13 namespace open3d {
14 namespace core {
15 namespace kernel {
16 
17 void IndexGet(const Tensor& src,
18  Tensor& dst,
19  const std::vector<Tensor>& index_tensors,
20  const SizeVector& indexed_shape,
21  const SizeVector& indexed_strides);
22 
23 void IndexGetCPU(const Tensor& src,
24  Tensor& dst,
25  const std::vector<Tensor>& index_tensors,
26  const SizeVector& indexed_shape,
27  const SizeVector& indexed_strides);
28 
29 #ifdef BUILD_CUDA_MODULE
30 void IndexGetCUDA(const Tensor& src,
31  Tensor& dst,
32  const std::vector<Tensor>& index_tensors,
33  const SizeVector& indexed_shape,
34  const SizeVector& indexed_strides);
35 #endif
36 
37 void IndexSet(const Tensor& src,
38  Tensor& dst,
39  const std::vector<Tensor>& index_tensors,
40  const SizeVector& indexed_shape,
41  const SizeVector& indexed_strides);
42 
43 void IndexSetCPU(const Tensor& src,
44  Tensor& dst,
45  const std::vector<Tensor>& index_tensors,
46  const SizeVector& indexed_shape,
47  const SizeVector& indexed_strides);
48 
49 #ifdef BUILD_CUDA_MODULE
50 void IndexSetCUDA(const Tensor& src,
51  Tensor& dst,
52  const std::vector<Tensor>& index_tensors,
53  const SizeVector& indexed_shape,
54  const SizeVector& indexed_strides);
55 #endif
56 
57 } // namespace kernel
58 } // namespace core
59 } // namespace open3d
void IndexGet(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition: IndexGetSet.cpp:21
void IndexSetCPU(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition: IndexGetSetCPU.cpp:61
void IndexSet(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition: IndexGetSet.cpp:47
void IndexGetCPU(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition: IndexGetSetCPU.cpp:41
Definition: PinholeCameraIntrinsic.cpp:16