container::op::stride_op Struct Reference

A functor to perform stride operation on a Tensor. More...

#include <linalg_op.h>

Collaboration diagram for container::op::stride_op:

Public Member Functions
void	operator() (const Tensor &input, const std::vector< int64_t > &stride, Tensor &output)
	Perform stride operation on the input Tensor.

Detailed Description

A functor to perform stride operation on a Tensor.

This functor applies a stride operation on the input Tensor, resulting in a new Tensor with data elements selected based on the specified stride. The stride specifies how elements are skipped in each dimension during the selection process.

Template Parameters

T	The data type of the Tensor.
Device	The execution device (e.g., CPU or GPU).

Member Function Documentation

operator()()

void container::op::stride_op::operator()	(	const Tensor &	input,
		const std::vector< int64_t > &	stride,
		Tensor &	output
	)

Perform stride operation on the input Tensor.

This function applies the stride operation on the input Tensor, generating a new Tensor with data elements selected based on the specified stride.

Parameters

input	The input Tensor on which the stride operation is performed.
stride	The stride specification, represented as a TensorShape. The stride specifies how elements are skipped in each dimension during the selection process.
output	The output Tensor that will hold the result of the stride operation. It must have the appropriate size to store the selected elements.

Here is the call graph for this function:

---

The documentation for this struct was generated from the following files:

• /home/runner/work/abacus-develop/abacus-develop/source/source_base/module_container/ATen/ops/linalg_op.h
• /home/runner/work/abacus-develop/abacus-develop/source/source_base/module_container/ATen/ops/linalg_op.cpp