TensorValue< T > Class Template Reference

Concrete definition of tensor value. More...

Detailed Description

template<typename T>
class neml2::TensorValue< T >

Concrete definition of tensor value.

#include <TensorValue.h>

Inheritance diagram for TensorValue< T >:

Public Member Functions
	TensorValue (T value)
void	to_ (const TensorOptions &options) override
	Send the value to the target options.
void	requires_grad_ (bool req=true) override
	Require grad.
	operator Tensor () const override
	Convert the parameter value to a Tensor.
void	operator= (const Tensor &val) override
	assignment operator
TensorType	type () const override
	Tensor type.
const T &	operator() () const
void	assign (const ATensor &val, TracerPrivilege key) override
	Secret assignment operator used by low-level operations such as jit tracing.
Public Member Functions inherited from TensorValueBase
	TensorValueBase ()=default
	TensorValueBase (const TensorValueBase &)=default
	TensorValueBase (TensorValueBase &&) noexcept=default
TensorValueBase &	operator= (const TensorValueBase &)=default
TensorValueBase &	operator= (TensorValueBase &&) noexcept=default
virtual	~TensorValueBase ()=default

Additional Inherited Members
Protected Attributes inherited from TensorValueBase
Size	_cached_intmd_dim = 0
	Cached intermediate dimension.

Constructor & Destructor Documentation

TensorValue()

template<typename T >

TensorValue ( T value )

inlineexplicit

Member Function Documentation

assign()

template<typename T >

void assign ( const ATensor & val, TracerPrivilege key )

overridevirtual

Secret assignment operator used by low-level operations such as jit tracing.

Implements TensorValueBase.

operator Tensor()

template<typename T >

operator Tensor ( ) const

overridevirtual

Convert the parameter value to a Tensor.

Implements TensorValueBase.

operator()()

template<typename T >

const T & operator() ( ) const

inline

operator=()

template<typename T >

void operator= ( const Tensor & val )

overridevirtual

assignment operator

Implements TensorValueBase.

requires_grad_()

template<typename T >

void requires_grad_ ( bool req = true )

overridevirtual

Require grad.

Implements TensorValueBase.

to_()

template<typename T >

void to_ ( const TensorOptions & )

overridevirtual

Send the value to the target options.

Implements TensorValueBase.

type()

template<typename T >

TensorType type ( ) const

overridevirtual

Tensor type.

Implements TensorValueBase.