NEML2 2.0.0
|
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).
#include <WWR4.h>
Public Member Functions | |
WWR4 (const R4 &F) | |
![]() | |
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 | |
WWR4 | variable_data () const |
WWR4 | clone () const |
WWR4 | detach () const |
Discard function graph. | |
WWR4 | to (const TensorOptions &options) const |
Change tensor options. | |
WWR4 | 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. | |
WWR4 | 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. | |
WWR4 | 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) |
WWR4 | batch_expand (const TraceableTensorShape &batch_shape) const |
WWR4 | 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. | |
WWR4 | 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. | |
WWR4 | 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. | |
WWR4 | batch_reshape (const TraceableTensorShape &batch_shape) const |
Reshape batch dimensions. | |
neml2::Tensor | base_reshape (TensorShapeRef base_shape) const |
Reshape base dimensions. | |
WWR4 | batch_unsqueeze (Size d) const |
Unsqueeze a batch dimension. | |
neml2::Tensor | base_unsqueeze (Size d) const |
Unsqueeze a base dimension. | |
WWR4 | 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 | |
static WWR4 | identity (const TensorOptions &options=default_tensor_options()) |
Create the identity tensor | |
![]() | |
static WWR4 | create (const TensorDataContainer &data, const TensorOptions &options=default_tensor_options()) |
Arbitrary tensor from a nested container with inferred batch dimension. | |
static WWR4 | empty (const TensorOptions &options=default_tensor_options()) |
Unbatched empty tensor. | |
static WWR4 | empty (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
Empty tensor given batch shape. | |
static WWR4 | zeros (const TensorOptions &options=default_tensor_options()) |
Unbatched zero tensor. | |
static WWR4 | zeros (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
Zero tensor given batch shape. | |
static WWR4 | ones (const TensorOptions &options=default_tensor_options()) |
Unbatched unit tensor. | |
static WWR4 | ones (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
Unit tensor given batch shape. | |
static WWR4 | full (Real init, const TensorOptions &options=default_tensor_options()) |
Unbatched tensor filled with a given value given base shape. | |
static WWR4 | 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 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, Real init) |
static WWR4 | linspace (const WWR4 &start, const WWR4 &end, Size nstep, Size dim=0) |
Create a new tensor by adding a new batch dimension with linear spacing between start and end . | |
static WWR4 | logspace (const WWR4 &start, const WWR4 &end, Size nstep, Size dim=0, Real base=10) |
log-space equivalent of the linspace named constructor | |
Additional Inherited Members | |
![]() | |
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. | |
|
static |
Create the identity tensor