container::op::inflate_op Struct Reference

A functor for inflating a tensor. More...

#include <linalg_op.h>

Collaboration diagram for container::op::inflate_op:

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

Detailed Description

A functor for inflating a tensor.

This struct defines a functor that can be used to inflate a tensor using the specified stride.

Member Function Documentation

operator()()

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

Inflate the input tensor.

This function inflates the input tensor using the given stride and stores the result in the output tensor.

Parameters

input	The input tensor to be inflated.
stride	The stride to use for inflation.
output	The output tensor where the inflated data will be stored.

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