tensor_converter.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <Optional.h>
 
#include "core/SizeVector.h"
#include "core/Tensor.h"
#include "pybind/cloudViewer_pybind.h"
 
namespace cloudViewer {
namespace core {
 
py::array TensorToPyArray(const Tensor& tensor);
 
Tensor PyArrayToTensor(py::array array, bool inplace);
 
Tensor PyListToTensor(const py::list& list,
                      utility::optional<Dtype> dtype = utility::nullopt,
                      utility::optional<Device> device = utility::nullopt);
 
Tensor PyTupleToTensor(const py::tuple& tuple,
                       utility::optional<Dtype> dtype = utility::nullopt,
                       utility::optional<Device> device = utility::nullopt);
 
Tensor DoubleToTensor(double scalar_value,
                      utility::optional<Dtype> dtype = utility::nullopt,
                      utility::optional<Device> device = utility::nullopt);
 
Tensor IntToTensor(int64_t scalar_value,
                   utility::optional<Dtype> dtype = utility::nullopt,
                   utility::optional<Device> device = utility::nullopt);
 
Tensor BoolToTensor(bool scalar_value,
                    utility::optional<Dtype> dtype = utility::nullopt,
                    utility::optional<Device> device = utility::nullopt);
 
Tensor PyHandleToTensor(const py::handle& handle,
                        utility::optional<Dtype> dtype = utility::nullopt,
                        utility::optional<Device> device = utility::nullopt,
                        bool force_copy = false);
 
}  // namespace core
}  // namespace cloudViewer