Open3D (C++ API)  0.16.1
TensorCheck.h
Go to the documentation of this file.
1// ----------------------------------------------------------------------------
2// - Open3D: www.open3d.org -
3// ----------------------------------------------------------------------------
4// The MIT License (MIT)
5//
6// Copyright (c) 2018-2021 www.open3d.org
7//
8// Permission is hereby granted, free of charge, to any person obtaining a copy
9// of this software and associated documentation files (the "Software"), to deal
10// in the Software without restriction, including without limitation the rights
11// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
12// copies of the Software, and to permit persons to whom the Software is
13// furnished to do so, subject to the following conditions:
14//
15// The above copyright notice and this permission notice shall be included in
16// all copies or substantial portions of the Software.
17//
18// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
19// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
20// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
21// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
22// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
23// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
24// IN THE SOFTWARE.
25// ----------------------------------------------------------------------------
26
27#pragma once
28
29#include "open3d/Macro.h"
30#include "open3d/core/Device.h"
31#include "open3d/core/Dtype.h"
32#include "open3d/core/Tensor.h"
33
40#define AssertTensorDtype(tensor, ...) \
41 tensor_check::AssertTensorDtype_( \
42 __FILE__, __LINE__, static_cast<const char*>(OPEN3D_FUNCTION), \
43 tensor, __VA_ARGS__)
44
51#define AssertTensorDtypes(tensor, ...) \
52 tensor_check::AssertTensorDtypes_( \
53 __FILE__, __LINE__, static_cast<const char*>(OPEN3D_FUNCTION), \
54 tensor, __VA_ARGS__)
55
62#define AssertTensorDevice(tensor, ...) \
63 tensor_check::AssertTensorDevice_( \
64 __FILE__, __LINE__, static_cast<const char*>(OPEN3D_FUNCTION), \
65 tensor, __VA_ARGS__)
66
77#define AssertTensorShape(tensor, ...) \
78 tensor_check::AssertTensorShape_( \
79 __FILE__, __LINE__, static_cast<const char*>(OPEN3D_FUNCTION), \
80 tensor, __VA_ARGS__)
81
82namespace open3d {
83namespace core {
84namespace tensor_check {
85
86void AssertTensorDtype_(const char* file,
87 int line,
88 const char* function,
89 const Tensor& tensor,
90 const Dtype& dtype);
91
92void AssertTensorDtypes_(const char* file,
93 int line,
94 const char* function,
95 const Tensor& tensor,
96 const std::vector<Dtype>& dtypes);
97
98void AssertTensorDevice_(const char* file,
99 int line,
100 const char* function,
101 const Tensor& tensor,
102 const Device& device);
103
104void AssertTensorShape_(const char* file,
105 int line,
106 const char* function,
107 const Tensor& tensor,
108 const DynamicSizeVector& shape);
109
110} // namespace tensor_check
111} // namespace core
112} // namespace open3d
void AssertTensorShape_(const char *file, int line, const char *function, const Tensor &tensor, const DynamicSizeVector &shape)
Definition: TensorCheck.cpp:90
void AssertTensorDtype_(const char *file, int line, const char *function, const Tensor &tensor, const Dtype &dtype)
Definition: TensorCheck.cpp:41
void AssertTensorDevice_(const char *file, int line, const char *function, const Tensor &tensor, const Device &device)
Definition: TensorCheck.cpp:76
void AssertTensorDtypes_(const char *file, int line, const char *function, const Tensor &tensor, const std::vector< Dtype > &dtypes)
Definition: TensorCheck.cpp:55
Definition: PinholeCameraIntrinsic.cpp:35