|
NEML2 2.0.0
|
Fourth order tensor without symmetry. More...
Fourth order tensor without symmetry.
The storage space is (3, 3, 3, 3).
#include <R4.h>

Public Member Functions | |
| R4 (const SSR4 &T) | |
| R4 (const WWR4 &T) | |
| R4 | rotate (const Rot &r) const |
| Rotate. | |
| R5 | drotate (const Rot &r) const |
| Derivative of the rotated tensor w.r.t. the Rodrigues vector. | |
| R8 | drotate_self (const Rot &r) const |
| Derivative of the rotated tensor w.r.t. itself. | |
| Scalar | operator() (Size i, Size j, Size k, Size l) const |
| Accessor. | |
| R4 | transpose (Size d1, Size d2) const |
| Arbitrary transpose two dimensions. | |
| R4 | transpose_minor () const |
| Transpose minor axes. | |
| R4 | transpose_major () const |
| Transpose major axes. | |
Public Member Functions inherited from PrimitiveTensor< R4, 3, 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< R4 > | |
| 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 (double)=delete | |
| TensorBase (float)=delete | |
| TensorBase (int)=delete | |
| R4 | variable_data () const |
| R4 | clone () const |
| R4 | detach () const |
| Discard function graph. | |
| R4 | to (const TensorOptions &options) const |
| Change tensor options. | |
| R4 | 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. | |
| R4 | 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. | |
| R4 | 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, const CScalar &v) |
| void | base_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other) |
| void | base_index_put_ (indexing::TensorIndicesRef indices, const CScalar &v) |
| R4 | batch_expand (const TraceableTensorShape &batch_shape) const |
| R4 | 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. | |
| R4 | 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. | |
| R4 | 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. | |
| R4 | batch_reshape (const TraceableTensorShape &batch_shape) const |
| Reshape batch dimensions. | |
| neml2::Tensor | base_reshape (TensorShapeRef base_shape) const |
| Reshape base dimensions. | |
| R4 | batch_unsqueeze (Size d) const |
| Unsqueeze a batch dimension. | |
| neml2::Tensor | base_unsqueeze (Size d) const |
| Unsqueeze a base dimension. | |
| R4 | 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. | |
Additional Inherited Members | |
Static Public Member Functions inherited from PrimitiveTensor< R4, 3, 3, 3, 3 > | |
| static R4 | create (const TensorDataContainer &data, const TensorOptions &options=default_tensor_options()) |
| Arbitrary tensor from a nested container with inferred batch dimension. | |
| static R4 | empty (const TensorOptions &options=default_tensor_options()) |
| Unbatched empty tensor. | |
| static R4 | empty (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
| Empty tensor given batch shape. | |
| static R4 | zeros (const TensorOptions &options=default_tensor_options()) |
| Unbatched zero tensor. | |
| static R4 | zeros (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
| Zero tensor given batch shape. | |
| static R4 | ones (const TensorOptions &options=default_tensor_options()) |
| Unbatched unit tensor. | |
| static R4 | ones (const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) |
| Unit tensor given batch shape. | |
| static R4 | full (const CScalar &init, const TensorOptions &options=default_tensor_options()) |
| Unbatched tensor filled with a given value given base shape. | |
| static R4 | full (const TraceableTensorShape &batch_shape, const CScalar &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< R4 > | |
| static R4 | empty_like (const R4 &other) |
| static R4 | zeros_like (const R4 &other) |
| Zero tensor like another, i.e. same batch and base shapes, same tensor options, etc. | |
| static R4 | ones_like (const R4 &other) |
| Unit tensor like another, i.e. same batch and base shapes, same tensor options, etc. | |
| static R4 | full_like (const R4 &other, const CScalar &init) |
| static R4 | linspace (const R4 &start, const R4 &end, Size nstep, Size dim=0) |
Create a new tensor by adding a new batch dimension with linear spacing between start and end. | |
| static R4 | logspace (const R4 &start, const R4 &end, Size nstep, Size dim=0, const CScalar &base=10) |
| log-space equivalent of the linspace named constructor | |
Static Public Attributes inherited from PrimitiveTensor< R4, 3, 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. | |
Protected Attributes inherited from TensorBase< R4 > | |
| TraceableTensorShape | _batch_sizes |
| Traceable batch sizes. | |
Derivative of the rotated tensor w.r.t. the Rodrigues vector.
Derivative of the rotated tensor w.r.t. itself.
| R4 transpose_major | ( | ) | const |
Transpose major axes.
| R4 transpose_minor | ( | ) | const |
Transpose minor axes.