Open3D (C++ API)  0.17.0
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"
12
13namespace open3d {
14namespace core {
15namespace kernel {
16
17void 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
23void 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
30void 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
37void 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
43void 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
50void 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