37 template <
class T,
class TIndex =
int32_t>
43 indices_ =
Tensor::Empty({int64_t(num)}, Dtype::FromType<TIndex>(),
50 Dtype::FromType<TIndex>(), device_);
61 distances_ =
Tensor::Full({int64_t(num)}, value, Dtype::FromType<T>(),
67 counts_ =
Tensor::Empty({int64_t(num)}, Dtype::FromType<TIndex>(),
73 counts_ =
Tensor::Full({int64_t(num)}, value, Dtype::FromType<TIndex>(),
T * GetDataPtr()
Definition: Tensor.h:1108
static Tensor Empty(const SizeVector &shape, Dtype dtype, const Device &device=Device("CPU:0"))
Create a tensor with uninitialized values.
Definition: Tensor.cpp:374
static Tensor Full(const SizeVector &shape, T fill_value, Dtype dtype, const Device &device=Device("CPU:0"))
Create a tensor fill with specified value.
Definition: Tensor.h:230
Definition: NeighborSearchAllocator.h:38
void AllocCounts(TIndex **ptr, size_t num, TIndex value)
Definition: NeighborSearchAllocator.h:72
const T * DistancesPtr() const
Definition: NeighborSearchAllocator.h:80
void AllocDistances(T **ptr, size_t num)
Definition: NeighborSearchAllocator.h:54
const TIndex * IndicesPtr() const
Definition: NeighborSearchAllocator.h:78
const Tensor & NeighborsIndex() const
Definition: NeighborSearchAllocator.h:84
void AllocDistances(T **ptr, size_t num, T value)
Definition: NeighborSearchAllocator.h:60
const Tensor & NeighborsDistance() const
Definition: NeighborSearchAllocator.h:86
void AllocIndices(TIndex **ptr, size_t num)
Definition: NeighborSearchAllocator.h:42
Tensor & NeighborsDistance_()
Definition: NeighborSearchAllocator.h:87
const Tensor & NeighborsCount() const
Definition: NeighborSearchAllocator.h:88
void AllocCounts(TIndex **ptr, size_t num)
Definition: NeighborSearchAllocator.h:66
NeighborSearchAllocator(Device device)
Definition: NeighborSearchAllocator.h:40
Tensor & NeighborsIndex_()
Definition: NeighborSearchAllocator.h:85
const TIndex * CountsPtr() const
Definition: NeighborSearchAllocator.h:82
void AllocIndices(TIndex **ptr, size_t num, TIndex value)
Definition: NeighborSearchAllocator.h:48
Definition: PinholeCameraIntrinsic.cpp:35