ShapeUtil.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2023 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include "open3d/core/Dispatch.h"
#include "open3d/core/SizeVector.h"
#include "open3d/utility/Logging.h"
 
namespace open3d {
namespace core {
 
class Tensor;
 
namespace shape_util {
 
bool IsCompatibleBroadcastShape(const SizeVector& l_shape,
                                const SizeVector& r_shape);
 
SizeVector BroadcastedShape(const SizeVector& l_shape,
                            const SizeVector& r_shape);
 
bool CanBeBrocastedToShape(const SizeVector& src_shape,
                           const SizeVector& dst_shape);
 
SizeVector ReductionShape(const SizeVector& src_shape,
                          const SizeVector& dims,
                          bool keepdim);
 
int64_t WrapDim(int64_t dim, int64_t max_dim, bool inclusive = false);
 
SizeVector InferShape(SizeVector shape, int64_t num_elements);
 
SizeVector Concat(const SizeVector& l_shape, const SizeVector& r_shape);
 
SizeVector Iota(int64_t n);
 
SizeVector DefaultStrides(const SizeVector& shape);
 
std::pair<bool, SizeVector> Restride(const SizeVector& old_shape,
                                     const SizeVector& old_strides,
                                     const SizeVector& new_shape);
 
}  // namespace shape_util
}  // namespace core
}  // namespace open3d