Open3D (C++ API)  0.17.0
Loading...
Searching...
No Matches
RaggedToDenseOpKernel.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 "torch/script.h"
11
12template <class T>
13torch::Tensor RaggedToDenseCPU(const torch::Tensor& values,
14 const torch::Tensor& row_splits,
15 const int64_t out_col_size,
16 const torch::Tensor& default_value);
17
18#ifdef BUILD_CUDA_MODULE
19template <class T>
20torch::Tensor RaggedToDenseCUDA(const torch::Tensor& values,
21 const torch::Tensor& row_splits,
22 const int64_t out_col_size,
23 const torch::Tensor& default_value);
24#endif
torch::Tensor RaggedToDenseCPU(const torch::Tensor &values, const torch::Tensor &row_splits, const int64_t out_col_size, const torch::Tensor &default_value)
Definition RaggedToDenseOpKernel.cpp:16