ShapeUtil.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 <Logging.h>
 
#include "cloudViewer/core/Dispatch.h"
#include "cloudViewer/core/SizeVector.h"
 
namespace cloudViewer {
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 cloudViewer