NEML2 2.0.0
|
The third order tensor with symmetry in the first two dimensions. More...
The third order tensor with symmetry in the first two dimensions.
Mandel notation is used for the first two symmetry dimensions, and so the storage space is (6, 3).
#include <SFR3.h>
Additional Inherited Members | |
![]() | |
PrimitiveTensor ()=default | |
Special member functions. | |
PrimitiveTensor (const ATensor &tensor, Size batch_dim) | |
Construct from another ATensor given batch dimension. | |
PrimitiveTensor (const ATensor &tensor, const TraceableTensorShape &batch_shape) | |
Construct from another ATensor given batch shape. | |
PrimitiveTensor (const Tensor &tensor) | |
Copy constructor. | |
PrimitiveTensor (const ATensor &tensor) | |
Construct from another ATensor and infer batch dimension. | |
operator Tensor () const | |
Implicit conversion to a Tensor and loses information on the fixed base shape. | |
![]() | |
TensorBase ()=default | |
Special member functions. | |
TensorBase (const ATensor &tensor, Size batch_dim) | |
Construct from another ATensor with given batch dimension. | |
TensorBase (const ATensor &tensor, const TraceableTensorShape &batch_shape) | |
Construct from another ATensor with given batch shape. | |
TensorBase (const neml2::Tensor &tensor) | |
Copy constructor. | |
TensorBase (Real)=delete | |
SFR3 | variable_data () const |
SFR3 | clone () const |
SFR3 | detach () const |
Discard function graph. | |
SFR3 | to (const TensorOptions &options) const |
Change tensor options. | |
SFR3 | operator- () const |
Negation. | |
bool | batched () const |
Whether the tensor is batched. | |
Size | batch_dim () const |
Return the number of batch dimensions. | |
Size | base_dim () const |
Return the number of base dimensions. | |
const TraceableTensorShape & | batch_sizes () const |
Return the batch size. | |
TraceableSize | batch_size (Size index) const |
Return the size of a batch axis. | |
TensorShapeRef | base_sizes () const |
Return the base size. | |
Size | base_size (Size index) const |
Return the size of a base axis. | |
Size | base_storage () const |
Return the flattened storage needed just for the base indices. | |
SFR3 | batch_index (indexing::TensorIndicesRef indices) const |
Get a tensor by slicing on the batch dimensions. | |
neml2::Tensor | base_index (indexing::TensorIndicesRef indices) const |
Get a tensor by slicing on the base dimensions. | |
SFR3 | batch_slice (Size dim, const indexing::Slice &index) const |
Get a tensor by slicing along a batch dimension. | |
neml2::Tensor | base_slice (Size dim, const indexing::Slice &index) const |
Get a tensor by slicing along a base dimension. | |
void | batch_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other) |
void | batch_index_put_ (indexing::TensorIndicesRef indices, Real v) |
void | base_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other) |
void | base_index_put_ (indexing::TensorIndicesRef indices, Real v) |
SFR3 | batch_expand (const TraceableTensorShape &batch_shape) const |
SFR3 | batch_expand (const TraceableSize &batch_size, Size dim) const |
Return a new view of the tensor with values broadcast along a given batch dimension. | |
neml2::Tensor | base_expand (TensorShapeRef base_shape) const |
Return a new view of the tensor with values broadcast along the base dimensions. | |
neml2::Tensor | base_expand (Size base_size, Size dim) const |
Return a new view of the tensor with values broadcast along a given base dimension. | |
SFR3 | batch_expand_as (const neml2::Tensor &other) const |
Expand the batch to have the same shape as another tensor. | |
neml2::Tensor | base_expand_as (const neml2::Tensor &other) const |
Expand the base to have the same shape as another tensor. | |
SFR3 | batch_expand_copy (const TraceableTensorShape &batch_shape) const |
Return a new tensor with values broadcast along the batch dimensions. | |
neml2::Tensor | base_expand_copy (TensorShapeRef base_shape) const |
Return a new tensor with values broadcast along the base dimensions. | |
SFR3 | batch_reshape (const TraceableTensorShape &batch_shape) const |
Reshape batch dimensions. | |
neml2::Tensor | base_reshape (TensorShapeRef base_shape) const |
Reshape base dimensions. | |
SFR3 | batch_unsqueeze (Size d) const |
Unsqueeze a batch dimension. | |
neml2::Tensor | base_unsqueeze (Size d) const |
Unsqueeze a base dimension. | |
SFR3 | batch_transpose (Size d1, Size d2) const |
Transpose two batch dimensions. | |
neml2::Tensor | base_transpose (Size d1, Size d2) const |
Transpose two base dimensions. | |
neml2::Tensor | base_flatten () const |
Flatten base dimensions. | |
![]() | |
static SFR3 | create (const TensorDataContainer &data, const TensorOptions &options=default_tensor_options()) |
Arbitrary tensor from a nested container with inferred batch dimension. | |
static SFR3 | empty (const TensorOptions &options=default_tensor_options()) |
Unbatched empty tensor. | |
static SFR3 | empty (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
Empty tensor given batch shape. | |
static SFR3 | zeros (const TensorOptions &options=default_tensor_options()) |
Unbatched zero tensor. | |
static SFR3 | zeros (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
Zero tensor given batch shape. | |
static SFR3 | ones (const TensorOptions &options=default_tensor_options()) |
Unbatched unit tensor. | |
static SFR3 | ones (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
Unit tensor given batch shape. | |
static SFR3 | full (Real init, const TensorOptions &options=default_tensor_options()) |
Unbatched tensor filled with a given value given base shape. | |
static SFR3 | full (const TraceableTensorShape &batch_shape, Real init, const TensorOptions &options=default_tensor_options()) |
Full tensor given batch shape. | |
static Tensor | identity_map (const TensorOptions &) |
Derived tensor classes should define identity_map where appropriate. | |
![]() | |
static SFR3 | empty_like (const SFR3 &other) |
static SFR3 | zeros_like (const SFR3 &other) |
Zero tensor like another, i.e. same batch and base shapes, same tensor options, etc. | |
static SFR3 | ones_like (const SFR3 &other) |
Unit tensor like another, i.e. same batch and base shapes, same tensor options, etc. | |
static SFR3 | full_like (const SFR3 &other, Real init) |
static SFR3 | linspace (const SFR3 &start, const SFR3 &end, Size nstep, Size dim=0) |
Create a new tensor by adding a new batch dimension with linear spacing between start and end . | |
static SFR3 | logspace (const SFR3 &start, const SFR3 &end, Size nstep, Size dim=0, Real base=10) |
log-space equivalent of the linspace named constructor | |
![]() | |
static const TensorShape | const_base_sizes |
The base shape. | |
static constexpr Size | const_base_dim |
The base dim. | |
static const Size | const_base_storage |
The base storage. | |