37template <
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>(),
static Tensor Empty(const SizeVector &shape, Dtype dtype, const Device &device=Device("CPU:0"))
Create a tensor with uninitialized values.
Definition: Tensor.cpp:374
T * GetDataPtr()
Definition: Tensor.h:1108
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
void AllocDistances(T **ptr, size_t num)
Definition: NeighborSearchAllocator.h:54
const TIndex * IndicesPtr() const
Definition: NeighborSearchAllocator.h:78
const Tensor & NeighborsCount() const
Definition: NeighborSearchAllocator.h:88
void AllocDistances(T **ptr, size_t num, T value)
Definition: NeighborSearchAllocator.h:60
Tensor & NeighborsDistance_()
Definition: NeighborSearchAllocator.h:87
void AllocIndices(TIndex **ptr, size_t num)
Definition: NeighborSearchAllocator.h:42
Tensor & NeighborsIndex_()
Definition: NeighborSearchAllocator.h:85
const Tensor & NeighborsDistance() const
Definition: NeighborSearchAllocator.h:86
void AllocCounts(TIndex **ptr, size_t num)
Definition: NeighborSearchAllocator.h:66
const T * DistancesPtr() const
Definition: NeighborSearchAllocator.h:80
NeighborSearchAllocator(Device device)
Definition: NeighborSearchAllocator.h:40
const Tensor & NeighborsIndex() const
Definition: NeighborSearchAllocator.h:84
void AllocIndices(TIndex **ptr, size_t num, TIndex value)
Definition: NeighborSearchAllocator.h:48
const TIndex * CountsPtr() const
Definition: NeighborSearchAllocator.h:82
Definition: PinholeCameraIntrinsic.cpp:35