|
NEML2 2.0.0
|
The fourth order tensor with skew symmetry in the first two dimensions. More...
The fourth order tensor with skew symmetry in the first two dimensions.
Mandel notation is used for the first two symmetry dimensions, and so the storage space is (3, 3, 3).
#include <WFR4.h>
Additional Inherited Members | |
 Public Member Functions inherited from PrimitiveTensor< WFR4, 3, 3, 3 > | |
| 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. | |
| Public Member Functions inherited from TensorBase< WFR4 > | |
| 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 | |
| WFR4 | variable_data () const |
| WFR4 | clone () const |
| WFR4 | detach () const |
| Discard function graph. | |
| WFR4 | to (const TensorOptions &options) const |
| Change tensor options. | |
| WFR4 | 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. | |
| WFR4 | 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. | |
| WFR4 | 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) |
| WFR4 | batch_expand (const TraceableTensorShape &batch_shape) const |
| WFR4 | 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. | |
| WFR4 | 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. | |
| WFR4 | 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. | |
| WFR4 | batch_reshape (const TraceableTensorShape &batch_shape) const |
| Reshape batch dimensions. | |
| neml2::Tensor | base_reshape (TensorShapeRef base_shape) const |
| Reshape base dimensions. | |
| WFR4 | batch_unsqueeze (Size d) const |
| Unsqueeze a batch dimension. | |
| neml2::Tensor | base_unsqueeze (Size d) const |
| Unsqueeze a base dimension. | |
| WFR4 | 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 Public Member Functions inherited from PrimitiveTensor< WFR4, 3, 3, 3 > | |
| static WFR4 | create (const TensorDataContainer &data, const TensorOptions &options=default_tensor_options()) |
| Arbitrary tensor from a nested container with inferred batch dimension. | |
| static WFR4 | empty (const TensorOptions &options=default_tensor_options()) |
| Unbatched empty tensor. | |
| static WFR4 | empty (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
| Empty tensor given batch shape. | |
| static WFR4 | zeros (const TensorOptions &options=default_tensor_options()) |
| Unbatched zero tensor. | |
| static WFR4 | zeros (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
| Zero tensor given batch shape. | |
| static WFR4 | ones (const TensorOptions &options=default_tensor_options()) |
| Unbatched unit tensor. | |
| static WFR4 | ones (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
| Unit tensor given batch shape. | |
| static WFR4 | full (Real init, const TensorOptions &options=default_tensor_options()) |
| Unbatched tensor filled with a given value given base shape. | |
| static WFR4 | 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 Public Member Functions inherited from TensorBase< WFR4 > | |
| static WFR4 | empty_like (const WFR4 &other) |
| static WFR4 | zeros_like (const WFR4 &other) |
| Zero tensor like another, i.e. same batch and base shapes, same tensor options, etc. | |
| static WFR4 | ones_like (const WFR4 &other) |
| Unit tensor like another, i.e. same batch and base shapes, same tensor options, etc. | |
| static WFR4 | full_like (const WFR4 &other, Real init) |
| static WFR4 | linspace (const WFR4 &start, const WFR4 &end, Size nstep, Size dim=0) |
Create a new tensor by adding a new batch dimension with linear spacing between start and end. | |
| static WFR4 | logspace (const WFR4 &start, const WFR4 &end, Size nstep, Size dim=0, Real base=10) |
| log-space equivalent of the linspace named constructor | |
| Static Public Attributes inherited from PrimitiveTensor< WFR4, 3, 3, 3 > | |
| 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. | |
