|
NEML2 2.0.0
|
The base class to allow us to set up a polymorphic container of Tensors. The concrete definitions will be templated on the actual tensor type. More...
The base class to allow us to set up a polymorphic container of Tensors. The concrete definitions will be templated on the actual tensor type.
#include <TensorValue.h>

Public Member Functions | |
| TensorValueBase ()=default | |
| TensorValueBase (const TensorValueBase &)=default | |
| TensorValueBase (TensorValueBase &&) noexcept=default | |
| TensorValueBase & | operator= (const TensorValueBase &)=default |
| TensorValueBase & | operator= (TensorValueBase &&) noexcept=default |
| virtual | ~TensorValueBase ()=default |
| virtual void | to_ (const TensorOptions &)=0 |
| Send the value to the target options. | |
| virtual void | requires_grad_ (bool req=true)=0 |
| Require grad. | |
| virtual | operator Tensor () const =0 |
| Convert the parameter value to a Tensor. | |
| virtual void | operator= (const Tensor &val)=0 |
| assignment operator | |
| virtual TensorType | type () const =0 |
| Tensor type. | |
| virtual void | assign (const ATensor &val, TracerPrivilege key)=0 |
| Secret assignment operator used by low-level operations such as jit tracing. | |
Protected Attributes | |
| Size | _cached_intmd_dim = 0 |
| Cached intermediate dimension. | |
|
default |
|
default |
|
defaultnoexcept |
|
virtualdefault |
|
pure virtual |
Secret assignment operator used by low-level operations such as jit tracing.
Implemented in TensorValue< T >.
|
pure virtual |
Convert the parameter value to a Tensor.
Implemented in TensorValue< T >.
|
pure virtual |
assignment operator
Implemented in TensorValue< T >.
|
default |
|
defaultnoexcept |
|
pure virtual |
Require grad.
Implemented in TensorValue< T >.
|
pure virtual |
Send the value to the target options.
Implemented in TensorValue< T >.
|
pure virtual |
Tensor type.
Implemented in TensorValue< T >.
|
protected |
Cached intermediate dimension.