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 torch::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. | |
|
default |
|
default |
|
defaultnoexcept |
|
virtualdefault |
Convert the parameter value to a Tensor.
Implemented in TensorValue< T >.
assignment operator
Implemented in TensorValue< T >.
|
default |
|
defaultnoexcept |
Require grad.
Implemented in TensorValue< T >.
Send the value to the target options.
Implemented in TensorValue< T >.
|
pure virtual |
Tensor type.
Implemented in TensorValue< T >.