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

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

Detailed Description

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

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

#include <WSR4.h>

Inheritance diagram for WSR4:

Public Member Functions

 WSR4 (const R4 &F)
 
- Public Member Functions inherited from PrimitiveTensor< WSR4, 3, 6 >
 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< WSR4 >
 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
 
WSR4 variable_data () const
 Variable data without function graph.
 
WSR4 contiguous () const
 
WSR4 clone () const
 Clone (take ownership)
 
WSR4 detach () const
 Discard function graph.
 
WSR4 to (const TensorOptions &options) const
 Change tensor options.
 
WSR4 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
 
WSR4 dynamic_index (indexing::TensorIndicesRef indices) const
 
WSR4 intmd_index (indexing::TensorIndicesRef indices) const
 
neml2::Tensor base_index (indexing::TensorIndicesRef indices) const
 
WSR4 dynamic_slice (Size d, const indexing::Slice &index) const
 
WSR4 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)
 
WSR4 dynamic_expand (const TraceableTensorShape &shape) const
 
WSR4 intmd_expand (TensorShapeRef shape) const
 
neml2::Tensor base_expand (TensorShapeRef shape) const
 
WSR4 batch_expand (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) const
 
neml2::Tensor static_expand (TensorShapeRef intmd_shape, TensorShapeRef base_shape) const
 
WSR4 dynamic_expand (const TraceableSize &size, Size d) const
 
WSR4 intmd_expand (Size size, Size d) const
 
neml2::Tensor base_expand (Size size, Size d) const
 
WSR4 dynamic_expand_as (const neml2::Tensor &other) const
 
WSR4 intmd_expand_as (const neml2::Tensor &other) const
 
neml2::Tensor base_expand_as (const neml2::Tensor &other) const
 
WSR4 batch_expand_as (const neml2::Tensor &other) const
 
neml2::Tensor static_expand_as (const neml2::Tensor &other) const
 
WSR4 dynamic_reshape (const TraceableTensorShape &shape) const
 
WSR4 intmd_reshape (TensorShapeRef shape) const
 
neml2::Tensor base_reshape (TensorShapeRef shape) const
 
WSR4 batch_reshape (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) const
 
neml2::Tensor static_reshape (TensorShapeRef intmd_shape, TensorShapeRef base_shape) const
 
WSR4 dynamic_squeeze (Size d) const
 
WSR4 intmd_squeeze (Size d) const
 
neml2::Tensor base_squeeze (Size d) const
 
WSR4 dynamic_unsqueeze (Size d, Size n=1) const
 
WSR4 intmd_unsqueeze (Size d, Size n=1) const
 
neml2::Tensor base_unsqueeze (Size d, Size n=1) const
 
WSR4 dynamic_transpose (Size d1, Size d2) const
 
WSR4 intmd_transpose (Size d1, Size d2) const
 
neml2::Tensor base_transpose (Size d1, Size d2) const
 
WSR4 dynamic_movedim (Size old_dim, Size new_dim) const
 
WSR4 intmd_movedim (Size old_dim, Size new_dim) const
 
neml2::Tensor base_movedim (Size old_dim, Size new_dim) const
 
WSR4 dynamic_flatten () const
 
WSR4 intmd_flatten () const
 
neml2::Tensor base_flatten () const
 
WSR4 batch_flatten () const
 Flatten batch dimensions.
 
neml2::Tensor static_flatten () const
 Flatten static dimensions.
 

Additional Inherited Members

- Public Types inherited from PrimitiveTensor< WSR4, 3, 6 >
using base_sizes_sequence
 Base shape sequence.
 
- Static Public Member Functions inherited from PrimitiveTensor< WSR4, 3, 6 >
static WSR4 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 WSR4 einsum (c10::string_view equation, TensorList tensors)
 Einstein summation along base dimensions.
 
static WSR4 empty (const TensorOptions &options=default_tensor_options())
 
static WSR4 empty (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())
 
static WSR4 zeros (const TensorOptions &options=default_tensor_options())
 
static WSR4 zeros (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())
 
static WSR4 ones (const TensorOptions &options=default_tensor_options())
 
static WSR4 ones (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())
 
static WSR4 full (const CScalar &init, const TensorOptions &options=default_tensor_options())
 
static WSR4 full (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, const CScalar &init, const TensorOptions &options=default_tensor_options())
 
static WSR4 rand (const TensorOptions &options=default_tensor_options())
 
static WSR4 rand (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, const TensorOptions &options=default_tensor_options())
 
static WSR4 fill (Args &&... args)
 
- Static Public Member Functions inherited from TensorBase< WSR4 >
static WSR4 empty_like (const WSR4 &other)
 
static WSR4 zeros_like (const WSR4 &other)
 Zero tensor like another, i.e. same batch and base shapes, same tensor options, etc.
 
static WSR4 ones_like (const WSR4 &other)
 Unit tensor like another, i.e. same batch and base shapes, same tensor options, etc.
 
static WSR4 full_like (const WSR4 &other, const CScalar &init)
 
static WSR4 rand_like (const WSR4 &other)
 
- Static Public Attributes inherited from PrimitiveTensor< WSR4, 3, 6 >
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< WSR4, 3, 6 >
void validate_shapes_and_dims () const
 Validate shapes and dimensions.
 
- Protected Member Functions inherited from TensorBase< WSR4 >
void validate_shapes_and_dims () const
 Validate shapes and dimensions.
 

Constructor & Destructor Documentation

◆ WSR4()

WSR4 ( const R4 & F)