vslice.hpp File Reference

VSlice utility functions. More...

#include <array>
#include <utils/linalg.hpp>
#include <utils/tensorarray.hpp>
#include <torch/torch.h>

Go to the source code of this file.

Namespaces
namespace	iganet
namespace	iganet::utils

Functions
template<bool transpose = false, std::size_t N>
auto	iganet::utils::VSlice (const utils::TensorArray< N > &index, const std::array< int64_t, N > &start_offset, const std::array< int64_t, N > &stop_offset, const std::array< int64_t, N - 1 > &leading_dim=make_array< int64_t, N - 1 >(1))
	Vectorized version of torch::indexing::Slice (see https://pytorch.org/cppdocs/notes/tensor_indexing.html)
template<bool transpose = false>
auto	iganet::utils::VSlice (torch::Tensor index, int64_t start_offset, int64_t stop_offset)
	Vectorized version of torch::indexing::Slice (see https://pytorch.org/cppdocs/notes/tensor_indexing.html)

Detailed Description

VSlice utility functions.

Author

Matthias Moller

Copyright

This file is part of the IgANet 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/.