IncliArray
Loading...
Searching...
No Matches
utils.h File Reference

Utility functions for shape and stride computations used in NDArray. More...

#include <vector>
Include dependency graph for utils.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

namespace  detail
 

Functions

std::vector< int > detail::_computeStrides (std::vector< int > newShape)
 Computes strides from a new shape.
 
std::vector< int > detail::_broadcastShape (std::vector< int > a, std::vector< int > b)
 Computes the broadcasted shape of a and b together.
 
std::vector< int > detail::_broadcastStrides (std::vector< int > originalShape, std::vector< int > originalStrides, std::vector< int > targetShape)
 Computes the broadcasted strides based on original shape and strides, and target shape.
 
int detail::_computeOffset (std::vector< int > index, std::vector< int > strides)
 Computes offset based on the index and strides.
 

Detailed Description

Utility functions for shape and stride computations used in NDArray.

This file contains helper functions within the detail namespace for computing strides, broadcasting shapes and strides, and calculating offsets.