holoscan::calc_strides

Beta
View as Markdown
void holoscan::calc_strides(
const DLTensor &tensor,
std::vector<int64_t> &strides,
bool to_num_elements = false
)

Fill strides from the given DLTensor object.

The following fields are used to fill strides:

  • ndim
  • shape
  • dtype

If tensor’s strides is nullptr, strides argument is filled with the calculated strides of the given DLTensor object. Otherwise, strides argument is filled with the given DLTensor object’s strides. strides vector would be resized to the size of ndim field of the given DLTensor object.

Parameters

tensor
const DLTensor &

DLTensor object that holds information to fill strides.

strides
std::vector<int64_t> &

Strides to fill.

to_num_elements
boolDefaults to false

If true, the strides in strides argument are in number of elements, not bytes (default: false).