buildEdges.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file buildEdges.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2022 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include <cstdint>

namespace at {
class Tensor;
}

namespace Acts {
namespace detail {

at::Tensor postprocessEdgeTensor(at::Tensor edges, bool removeSelfLoops = true,
                                 bool removeDuplicates = true,
                                 bool flipDirections = false);

at::Tensor buildEdgesFRNN(at::Tensor& embedFeatures, float rVal, int kVal,
                          bool flipDirections = false);

at::Tensor buildEdgesKDTree(at::Tensor& embedFeatures, float rVal, int kVal,
                            bool flipDirections = false);

at::Tensor buildEdges(at::Tensor& embedFeatures, float rVal, int kVal,
                      bool flipDirections = false);

}  // namespace detail
}  // namespace Acts