The symmetric fourth order tensor, with skew symmetry in the first two dimensionss and symmetry in the second two.
More...
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).
|
| | WSR4 (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 |
| |
| 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 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 |
| |
| 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.
|
| |
|
| using | base_sizes_sequence |
| | Base shape sequence.
|
| |
| 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 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 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.
|
| |
| void | validate_shapes_and_dims () const |
| | Validate shapes and dimensions.
|
| |
| void | validate_shapes_and_dims () const |
| | Validate shapes and dimensions.
|
| |