NEML2 1.4.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 | |
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. | |
|
virtualdefault |
Convert the parameter value to a Tensor.
Implemented in TensorValue< T >.
assignment operator
Implemented in TensorValue< T >.
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 >.