NEML2 2.0.0
Loading...
Searching...
No Matches
SWR4 Class Reference

The symmetric fourth order tensor, with symmetry in the first two dimensionss and skew-symmetry in the second two. More...

Detailed Description

The symmetric fourth order tensor, with symmetry in the first two dimensionss and skew-symmetry in the second two.

Mandel notation/reduced skew is used, and so the storage space is (6, 3).

#include <SWR4.h>

Inheritance diagram for SWR4:

Public Member Functions

 SWR4 (const R4 &F)
 
- Public Member Functions inherited from PrimitiveTensor< SWR4, 6, 3 >
 PrimitiveTensor ()=default
 Special member functions.
 
 PrimitiveTensor (const ATensor &tensor, Size intmd_dim)
 Construct from an ATensor and infer dynamic shape.
 
 PrimitiveTensor (const ATensor &tensor, Size dynamic_dim, Size intmd_dim)
 Construct from an ATensor and extract dynamic shape given dynamic dimension.
 
 PrimitiveTensor (const ATensor &tensor, const TraceableTensorShape &dynamic_shape, Size intmd_dim)
 Construct from an ATensor given dynamic shape.
 
 PrimitiveTensor (const TensorBase< Derived2 > &tensor)
 Copy constructor.
 
 operator neml2::Tensor () const
 Implicit conversion to a Tensor (discards information on the fixed base shape)
 
Scalar operator() (Args... i) const
 Single-element accessor.
 
- Public Member Functions inherited from TensorBase< SWR4 >
 TensorBase ()=default
 Default constructor.
 
 TensorBase (const ATensor &tensor, Size dynamic_dim, Size intmd_dim)
 Construct from an ATensor with given dynamic dimension.
 
 TensorBase (const ATensor &tensor, TraceableTensorShape dynamic_shape, Size intmd_dim)
 Construct from an ATensor with given dynamic shape.
 
 TensorBase (const TensorBase< Derived2 > &tensor)
 Copy constructor.
 
 TensorBase (double)=delete
 
 TensorBase (float)=delete
 
 TensorBase (int)=delete
 
SWR4 variable_data () const
 Variable data without function graph.
 
SWR4 contiguous () const
 
SWR4 clone () const
 Clone (take ownership)
 
SWR4 detach () const
 Discard function graph.
 
SWR4 to (const TensorOptions &options) const
 Change tensor options.
 
SWR4 operator- () const
 Negation.
 
Size batch_dim () const
 
Size base_dim () const
 
Size dynamic_dim () const
 
Size static_dim () const
 
Size intmd_dim () const
 
TraceableTensorShape batch_sizes () const
 
TensorShapeRef base_sizes () const
 
const TraceableTensorShapedynamic_sizes () const
 
TensorShapeRef static_sizes () const
 
TensorShapeRef intmd_sizes () const
 
TraceableSize batch_size (Size i) const
 
Size base_size (Size i) const
 
const TraceableSizedynamic_size (Size i) const
 
Size static_size (Size i) const
 
Size intmd_size (Size i) const
 
SWR4 dynamic_index (indexing::TensorIndicesRef indices) const
 
SWR4 intmd_index (indexing::TensorIndicesRef indices) const
 
neml2::Tensor base_index (indexing::TensorIndicesRef indices) const
 
SWR4 dynamic_slice (Size d, const indexing::Slice &index) const
 
SWR4 intmd_slice (Size d, const indexing::Slice &index) const
 
neml2::Tensor base_slice (Size d, const indexing::Slice &index) const
 
void dynamic_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other)
 
void dynamic_index_put_ (indexing::TensorIndicesRef indices, const CScalar &v)
 
void intmd_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other)
 
void intmd_index_put_ (indexing::TensorIndicesRef indices, const CScalar &v)
 
void base_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other)
 
void base_index_put_ (indexing::TensorIndicesRef indices, const CScalar &v)
 
SWR4 dynamic_expand (const TraceableTensorShape &shape) const
 
SWR4 intmd_expand (TensorShapeRef shape) const
 
neml2::Tensor base_expand (TensorShapeRef shape) const
 
SWR4 batch_expand (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) const
 
neml2::Tensor static_expand (TensorShapeRef intmd_shape, TensorShapeRef base_shape) const
 
SWR4 dynamic_expand (const TraceableSize &size, Size d) const
 
SWR4 intmd_expand (Size size, Size d) const
 
neml2::Tensor base_expand (Size size, Size d) const
 
SWR4 dynamic_expand_as (const neml2::Tensor &other) const
 
SWR4 intmd_expand_as (const neml2::Tensor &other) const
 
neml2::Tensor base_expand_as (const neml2::Tensor &other) const
 
SWR4 batch_expand_as (const neml2::Tensor &other) const
 
neml2::Tensor static_expand_as (const neml2::Tensor &other) const
 
SWR4 dynamic_reshape (const TraceableTensorShape &shape) const
 
SWR4 intmd_reshape (TensorShapeRef shape) const
 
neml2::Tensor base_reshape (TensorShapeRef shape) const
 
SWR4 batch_reshape (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) const
 
neml2::Tensor static_reshape (TensorShapeRef intmd_shape, TensorShapeRef base_shape) const
 
SWR4 dynamic_squeeze (Size d) const
 
SWR4 intmd_squeeze (Size d) const
 
neml2::Tensor base_squeeze (Size d) const
 
SWR4 dynamic_unsqueeze (Size d, Size n=1) const
 
SWR4 intmd_unsqueeze (Size d, Size n=1) const
 
neml2::Tensor base_unsqueeze (Size d, Size n=1) const
 
SWR4 dynamic_transpose (Size d1, Size d2) const
 
SWR4 intmd_transpose (Size d1, Size d2) const
 
neml2::Tensor base_transpose (Size d1, Size d2) const
 
SWR4 dynamic_movedim (Size old_dim, Size new_dim) const
 
SWR4 intmd_movedim (Size old_dim, Size new_dim) const
 
neml2::Tensor base_movedim (Size old_dim, Size new_dim) const
 
SWR4 dynamic_flatten () const
 
SWR4 intmd_flatten () const
 
neml2::Tensor base_flatten () const
 
SWR4 batch_flatten () const
 Flatten batch dimensions.
 
neml2::Tensor static_flatten () const
 Flatten static dimensions.
 

Additional Inherited Members

- Public Types inherited from PrimitiveTensor< SWR4, 6, 3 >
using base_sizes_sequence
 Base shape sequence.
 
- Static Public Member Functions inherited from PrimitiveTensor< SWR4, 6, 3 >
static SWR4 create (const TensorDataContainer &data, Size intmd_dim=0, const TensorOptions &options=default_tensor_options())
 Arbitrary tensor from a nested container with inferred batch dimension.
 
static SWR4 einsum (c10::string_view equation, TensorList tensors)
 Einstein summation along base dimensions.
 
static SWR4 empty (const TensorOptions &options=default_tensor_options())
 
static SWR4 empty (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())
 
static SWR4 zeros (const TensorOptions &options=default_tensor_options())
 
static SWR4 zeros (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())
 
static SWR4 ones (const TensorOptions &options=default_tensor_options())
 
static SWR4 ones (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())
 
static SWR4 full (const CScalar &init, const TensorOptions &options=default_tensor_options())
 
static SWR4 full (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, const CScalar &init, const TensorOptions &options=default_tensor_options())
 
static SWR4 rand (const TensorOptions &options=default_tensor_options())
 
static SWR4 rand (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, const TensorOptions &options=default_tensor_options())
 
static SWR4 fill (Args &&... args)
 
- Static Public Member Functions inherited from TensorBase< SWR4 >
static SWR4 empty_like (const SWR4 &other)
 
static SWR4 zeros_like (const SWR4 &other)
 Zero tensor like another, i.e. same batch and base shapes, same tensor options, etc.
 
static SWR4 ones_like (const SWR4 &other)
 Unit tensor like another, i.e. same batch and base shapes, same tensor options, etc.
 
static SWR4 full_like (const SWR4 &other, const CScalar &init)
 
static SWR4 rand_like (const SWR4 &other)
 
- Static Public Attributes inherited from PrimitiveTensor< SWR4, 6, 3 >
static const TensorShape const_base_sizes
 The base shape.
 
static constexpr Size const_base_dim
 The base dim.
 
static constexpr Size const_base_numel
 The base numel.
 
- Protected Member Functions inherited from PrimitiveTensor< SWR4, 6, 3 >
void validate_shapes_and_dims () const
 Validate shapes and dimensions.
 
- Protected Member Functions inherited from TensorBase< SWR4 >
void validate_shapes_and_dims () const
 Validate shapes and dimensions.
 

Constructor & Destructor Documentation

◆ SWR4()

SWR4 ( const R4 & F)