|
NEML2 2.0.0
|
This is the complete list of members for WR2, including all inherited members.
| _batch_sizes | TensorBase< WR2 > | protected |
| base_dim() const | TensorBase< WR2 > | |
| base_expand(TensorShapeRef base_shape) const | TensorBase< WR2 > | |
| base_expand(Size base_size, Size dim) const | TensorBase< WR2 > | |
| base_expand_as(const neml2::Tensor &other) const | TensorBase< WR2 > | |
| base_expand_copy(TensorShapeRef base_shape) const | TensorBase< WR2 > | |
| base_flatten() const | TensorBase< WR2 > | |
| base_index(indexing::TensorIndicesRef indices) const | TensorBase< WR2 > | |
| base_index_put_(indexing::TensorIndicesRef indices, const ATensor &other) | TensorBase< WR2 > | |
| base_index_put_(indexing::TensorIndicesRef indices, const CScalar &v) | TensorBase< WR2 > | |
| base_reshape(TensorShapeRef base_shape) const | TensorBase< WR2 > | |
| base_size(Size index) const | TensorBase< WR2 > | |
| base_sizes() const | TensorBase< WR2 > | |
| base_slice(Size dim, const indexing::Slice &index) const | TensorBase< WR2 > | |
| base_storage() const | TensorBase< WR2 > | |
| base_transpose(Size d1, Size d2) const | TensorBase< WR2 > | |
| base_unsqueeze(Size d) const | TensorBase< WR2 > | |
| batch_dim() const | TensorBase< WR2 > | |
| batch_expand(const TraceableTensorShape &batch_shape) const | TensorBase< WR2 > | |
| batch_expand(const TraceableSize &batch_size, Size dim) const | TensorBase< WR2 > | |
| batch_expand_as(const neml2::Tensor &other) const | TensorBase< WR2 > | |
| batch_expand_copy(const TraceableTensorShape &batch_shape) const | TensorBase< WR2 > | |
| batch_index(indexing::TensorIndicesRef indices) const | TensorBase< WR2 > | |
| batch_index_put_(indexing::TensorIndicesRef indices, const ATensor &other) | TensorBase< WR2 > | |
| batch_index_put_(indexing::TensorIndicesRef indices, const CScalar &v) | TensorBase< WR2 > | |
| batch_reshape(const TraceableTensorShape &batch_shape) const | TensorBase< WR2 > | |
| batch_size(Size index) const | TensorBase< WR2 > | |
| batch_sizes() const | TensorBase< WR2 > | |
| batch_slice(Size dim, const indexing::Slice &index) const | TensorBase< WR2 > | |
| batch_transpose(Size d1, Size d2) const | TensorBase< WR2 > | |
| batch_unsqueeze(Size d) const | TensorBase< WR2 > | |
| batched() const | TensorBase< WR2 > | |
| clone() const | TensorBase< WR2 > | |
| const_base_dim | PrimitiveTensor< WR2, 3 > | static |
| const_base_sizes | PrimitiveTensor< WR2, 3 > | inlinestatic |
| const_base_storage | PrimitiveTensor< WR2, 3 > | inlinestatic |
| create(const TensorDataContainer &data, const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| detach() const | TensorBase< WR2 > | |
| dexp() const | WR2 | |
| drotate(const Rot &r) const | VecBase< WR2 > | |
| drotate(const R2 &R) const | VecBase< WR2 > | |
| empty(const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| empty(const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| empty_like(const WR2 &other) | TensorBase< WR2 > | static |
| exp() const | WR2 | |
| fill(const CScalar &v1, const CScalar &v2, const CScalar &v3, const TensorOptions &options=default_tensor_options()) | VecBase< WR2 > | static |
| fill(const Scalar &v1, const Scalar &v2, const Scalar &v3) | VecBase< WR2 > | static |
| full(const CScalar &init, const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| full(const TraceableTensorShape &batch_shape, const CScalar &init, const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| full_like(const WR2 &other, const CScalar &init) | TensorBase< WR2 > | static |
| identity_map(const TensorOptions &options=default_tensor_options()) | VecBase< WR2 > | static |
| linspace(const WR2 &start, const WR2 &end, Size nstep, Size dim=0) | TensorBase< WR2 > | static |
| logspace(const WR2 &start, const WR2 &end, Size nstep, Size dim=0, const CScalar &base=10) | TensorBase< WR2 > | static |
| norm() const | VecBase< WR2 > | |
| norm_sq() const | VecBase< WR2 > | |
| ones(const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| ones(const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| ones_like(const WR2 &other) | TensorBase< WR2 > | static |
| operator Tensor() const | PrimitiveTensor< WR2, 3 > | |
| operator()(Size i, Size j) const | WR2 | |
| neml2::VecBase< WR2 >::operator()(Size i) const | VecBase< WR2 > | |
| operator-() const | TensorBase< WR2 > | |
| PrimitiveTensor()=default | PrimitiveTensor< WR2, 3 > | |
| PrimitiveTensor(const ATensor &tensor, Size batch_dim) | PrimitiveTensor< WR2, 3 > | |
| PrimitiveTensor(const ATensor &tensor, const TraceableTensorShape &batch_shape) | PrimitiveTensor< WR2, 3 > | |
| PrimitiveTensor(const Tensor &tensor) | PrimitiveTensor< WR2, 3 > | |
| PrimitiveTensor(const ATensor &tensor) | PrimitiveTensor< WR2, 3 > | explicit |
| rotate(const Rot &r) const | VecBase< WR2 > | |
| rotate(const R2 &R) const | VecBase< WR2 > | |
| TensorBase()=default | TensorBase< WR2 > | |
| TensorBase(const ATensor &tensor, Size batch_dim) | TensorBase< WR2 > | |
| TensorBase(const ATensor &tensor, const TraceableTensorShape &batch_shape) | TensorBase< WR2 > | |
| TensorBase(const neml2::Tensor &tensor) | TensorBase< WR2 > | |
| TensorBase(double)=delete | TensorBase< WR2 > | |
| TensorBase(float)=delete | TensorBase< WR2 > | |
| TensorBase(int)=delete | TensorBase< WR2 > | |
| to(const TensorOptions &options) const | TensorBase< WR2 > | |
| variable_data() const | TensorBase< WR2 > | |
| WR2(const R2 &T) | WR2 | |
| zeros(const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| zeros(const TraceableTensorShape &batch_shape, const TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WR2, 3 > | static |
| zeros_like(const WR2 &other) | TensorBase< WR2 > | static |