Traceable size.
More...
Traceable size.
Similar to neml2::TraceableTensorShape, but only for a single dimension.
- See also
- neml2::TraceableTensorShape
#include <types.h>
◆ as_tensor()
torch::Tensor as_tensor |
( |
| ) |
const |
- Returns
- the size represented as a scalar tensor (possibly traceable)
◆ concrete()
- Returns
- the concrete size (without any traceable information)
◆ traceable()
const torch::Tensor * traceable |
( |
| ) |
const |
|
noexcept |
- Returns
- a pointer to the torch::Tensor representing the traceable size if it is traceable, otherwise a nullptr