tvm::te::Tensor Class Reference

Tensor structure representing a possible input, or intermediate computation result. More...

#include <tensor.h>

Inheritance diagram for tvm::te::Tensor:

Collaboration diagram for tvm::te::Tensor:

Classes
class	Slice
	data structure to represent a slice that fixes first k coordinates. This is used to enable syntax sugar of Tensor[x][y][z] to get the element. More...

Public Member Functions
	Tensor (Array< PrimExpr > shape, DataType dtype, Operation op, int value_index)
bool	operator== (const Tensor &other) const
	check if two tensors equals each other. More...
bool	operator!= (const Tensor &other) const
	check if two tensors are different. More...
size_t	ndim () const
template<typename... Args>
PrimExpr	operator() (Args &&... args) const
	Take elements from the tensor. More...
PrimExpr	operator() (Array< PrimExpr > indices) const
	Take elements from the tensor. More...
PrimExpr	operator() (Array< Var > indices) const
	Take elements from the tensor. More...
template<typename... Args>
PrimExpr	IndexWithNegativeIndices (Args &&... args) const
	Take elements from the tensor with support for negative indices. More...
PrimExpr	IndexWithNegativeIndices (Array< PrimExpr > indices) const
	Take elements from the tensor with support for negative indices. More...
PrimExpr	IndexWithNegativeIndices (Array< Var > indices) const
	Take elements from the tensor with support for negative indices. More...
Slice	operator[] (PrimExpr i) const
	get i-th slice from the current Tensor. More...
	TVM_DEFINE_OBJECT_REF_METHODS (Tensor, DataProducer, TensorNode)
Public Member Functions inherited from tvm::tir::DataProducer
	TVM_DEFINE_OBJECT_REF_METHODS (DataProducer, PrimExprConvertible, DataProducerNode)
Public Member Functions inherited from tvm::PrimExprConvertible
	TVM_DEFINE_OBJECT_REF_METHODS (PrimExprConvertible, ObjectRef, PrimExprConvertibleNode)

Detailed Description

Tensor structure representing a possible input, or intermediate computation result.

Constructor & Destructor Documentation

Tensor()

tvm::te::Tensor::Tensor	(	Array< PrimExpr >	shape,
		DataType	dtype,
		Operation	op,
		int	value_index
	)

Member Function Documentation

IndexWithNegativeIndices() [1/3]

template<typename... Args>

PrimExpr tvm::te::Tensor::IndexWithNegativeIndices ( Args &&...  args ) const

inline

Take elements from the tensor with support for negative indices.

Parameters

args	The indices

Returns

the result expression representing tensor read.

IndexWithNegativeIndices() [2/3]

PrimExpr tvm::te::Tensor::IndexWithNegativeIndices

(

Array< PrimExpr >

indices

)

const

Take elements from the tensor with support for negative indices.

Parameters

indices

the indices.

Returns

the result expression representing tensor read.

IndexWithNegativeIndices() [3/3]

PrimExpr tvm::te::Tensor::IndexWithNegativeIndices

(

Array< Var >

indices

)

const

Take elements from the tensor with support for negative indices.

Parameters

indices

the indices.

Returns

the result expression representing tensor read.

ndim()

size_t tvm::te::Tensor::ndim ( ) const

inline

Returns

The dimension of the tensor

operator!=()

bool tvm::te::Tensor::operator!= ( const Tensor &  other ) const

inline

check if two tensors are different.

Parameters

other

tensor to be checked.

Returns

whether the two tensors are different.

operator()() [1/3]

template<typename... Args>

PrimExpr tvm::te::Tensor::operator() ( Args &&...  args ) const

inline

Take elements from the tensor.

Parameters

args	The indices

Returns

the result expression representing tensor read.

operator()() [2/3]

PrimExpr tvm::te::Tensor::operator()

(

Array< PrimExpr >

indices

)

const

Take elements from the tensor.

Parameters

indices

the indices.

Returns

the result expression representing tensor read.

operator()() [3/3]

PrimExpr tvm::te::Tensor::operator()

(

Array< Var >

indices

)

const

Take elements from the tensor.

Parameters

indices

the indices.

Returns

the result expression representing tensor read.

operator==()

bool tvm::te::Tensor::operator== ( const Tensor &  other ) const

inline

check if two tensors equals each other.

Parameters

other

tensor to be checked.

Returns

whether the two tensors equals each other.

operator[]()

Slice tvm::te::Tensor::operator[] ( PrimExpr  i ) const

inline

get i-th slice from the current Tensor.

Parameters

i	the index of the coordinate

Returns

the subsequent slice.

TVM_DEFINE_OBJECT_REF_METHODS()

tvm::te::Tensor::TVM_DEFINE_OBJECT_REF_METHODS	(	Tensor	,
		DataProducer	,
		TensorNode
	)

---

The documentation for this class was generated from the following file:

• include/tvm/te/tensor.h