The symmetric fourth order tensor, with skew symmetry in the first two dimensions and last two dimensions.
More...
The symmetric fourth order tensor, with skew symmetry in the first two dimensions and last two dimensions.
Mandel notation/reduced skew is used, and so the storage space is (3, 3).
|
| | WWR4 (const R4 &F) |
| |
| | 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.
|
| |
| | 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 |
| |
| WWR4 | variable_data () const |
| | Variable data without function graph.
|
| |
| WWR4 | contiguous () const |
| |
| WWR4 | clone () const |
| | Clone (take ownership)
|
| |
| WWR4 | detach () const |
| | Discard function graph.
|
| |
| WWR4 | to (const TensorOptions &options) const |
| | Change tensor options.
|
| |
| WWR4 | 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 TraceableTensorShape & | dynamic_sizes () const |
| |
| TensorShapeRef | static_sizes () const |
| |
| TensorShapeRef | intmd_sizes () const |
| |
| TraceableSize | batch_size (Size i) const |
| |
| Size | base_size (Size i) const |
| |
| const TraceableSize & | dynamic_size (Size i) const |
| |
| Size | static_size (Size i) const |
| |
| Size | intmd_size (Size i) const |
| |
| WWR4 | dynamic_index (indexing::TensorIndicesRef indices) const |
| |
| WWR4 | intmd_index (indexing::TensorIndicesRef indices) const |
| |
| neml2::Tensor | base_index (indexing::TensorIndicesRef indices) const |
| |
| WWR4 | dynamic_slice (Size d, const indexing::Slice &index) const |
| |
| WWR4 | 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) |
| |
| WWR4 | dynamic_expand (const TraceableTensorShape &shape) const |
| |
| WWR4 | intmd_expand (TensorShapeRef shape) const |
| |
| neml2::Tensor | base_expand (TensorShapeRef shape) const |
| |
| WWR4 | batch_expand (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) const |
| |
| neml2::Tensor | static_expand (TensorShapeRef intmd_shape, TensorShapeRef base_shape) const |
| |
| WWR4 | dynamic_expand (const TraceableSize &size, Size d) const |
| |
| WWR4 | intmd_expand (Size size, Size d) const |
| |
| neml2::Tensor | base_expand (Size size, Size d) const |
| |
| WWR4 | dynamic_expand_as (const neml2::Tensor &other) const |
| |
| WWR4 | intmd_expand_as (const neml2::Tensor &other) const |
| |
| neml2::Tensor | base_expand_as (const neml2::Tensor &other) const |
| |
| WWR4 | batch_expand_as (const neml2::Tensor &other) const |
| |
| neml2::Tensor | static_expand_as (const neml2::Tensor &other) const |
| |
| WWR4 | dynamic_reshape (const TraceableTensorShape &shape) const |
| |
| WWR4 | intmd_reshape (TensorShapeRef shape) const |
| |
| neml2::Tensor | base_reshape (TensorShapeRef shape) const |
| |
| WWR4 | batch_reshape (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) const |
| |
| neml2::Tensor | static_reshape (TensorShapeRef intmd_shape, TensorShapeRef base_shape) const |
| |
| WWR4 | dynamic_squeeze (Size d) const |
| |
| WWR4 | intmd_squeeze (Size d) const |
| |
| neml2::Tensor | base_squeeze (Size d) const |
| |
| WWR4 | dynamic_unsqueeze (Size d, Size n=1) const |
| |
| WWR4 | intmd_unsqueeze (Size d, Size n=1) const |
| |
| neml2::Tensor | base_unsqueeze (Size d, Size n=1) const |
| |
| WWR4 | dynamic_transpose (Size d1, Size d2) const |
| |
| WWR4 | intmd_transpose (Size d1, Size d2) const |
| |
| neml2::Tensor | base_transpose (Size d1, Size d2) const |
| |
| WWR4 | dynamic_movedim (Size old_dim, Size new_dim) const |
| |
| WWR4 | intmd_movedim (Size old_dim, Size new_dim) const |
| |
| neml2::Tensor | base_movedim (Size old_dim, Size new_dim) const |
| |
| WWR4 | dynamic_flatten () const |
| |
| WWR4 | intmd_flatten () const |
| |
| neml2::Tensor | base_flatten () const |
| |
| WWR4 | batch_flatten () const |
| | Flatten batch dimensions.
|
| |
| neml2::Tensor | static_flatten () const |
| | Flatten static dimensions.
|
| |
|
| static WWR4 | identity (const TensorOptions &options=default_tensor_options()) |
| | Create the identity tensor \(\delta_{ij}\delta_{kl}\).
|
| |
| static WWR4 | 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 WWR4 | einsum (c10::string_view equation, TensorList tensors) |
| | Einstein summation along base dimensions.
|
| |
| static WWR4 | empty (const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | empty (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | zeros (const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | zeros (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | ones (const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | ones (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | full (const CScalar &init, const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | full (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, const CScalar &init, const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | rand (const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | rand (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, const TensorOptions &options=default_tensor_options()) |
| |
| static WWR4 | fill (Args &&... args) |
| |
| static WWR4 | empty_like (const WWR4 &other) |
| |
| static WWR4 | zeros_like (const WWR4 &other) |
| | Zero tensor like another, i.e. same batch and base shapes, same tensor options, etc.
|
| |
| static WWR4 | ones_like (const WWR4 &other) |
| | Unit tensor like another, i.e. same batch and base shapes, same tensor options, etc.
|
| |
| static WWR4 | full_like (const WWR4 &other, const CScalar &init) |
| |
| static WWR4 | rand_like (const WWR4 &other) |
| |