- Generated by
1.10.0
|
NEML2 1.4.0
|
This is the complete list of members for WR2, including all inherited members.
| base_dim() const | BatchTensorBase< Derived > | |
| base_expand(TorchShapeRef base_size) const | BatchTensorBase< Derived > | |
| base_expand_as(const Derived2 &other) const | BatchTensorBase< Derived > | |
| base_expand_copy(TorchShapeRef base_size) const | BatchTensorBase< Derived > | |
| base_index(const TorchSlice &indices) const | BatchTensorBase< Derived > | |
| base_index_put(const TorchSlice &indices, const torch::Tensor &other) | BatchTensorBase< Derived > | |
| base_movedim(TorchSize d1, TorchSize d2) const | BatchTensorBase< Derived > | |
| base_reshape(TorchShapeRef base_shape) const | BatchTensorBase< Derived > | |
| base_size(TorchSize index) const | BatchTensorBase< Derived > | |
| base_sizes() const | BatchTensorBase< Derived > | |
| base_storage() const | BatchTensorBase< Derived > | |
| base_transpose(TorchSize d1, TorchSize d2) const | BatchTensorBase< Derived > | |
| base_unsqueeze(TorchSize d) const | BatchTensorBase< Derived > | |
| batch_dim() const | BatchTensorBase< Derived > | |
| batch_dim() | BatchTensorBase< Derived > | |
| batch_expand(TorchShapeRef batch_size) const | BatchTensorBase< Derived > | |
| batch_expand_as(const Derived2 &other) const | BatchTensorBase< Derived > | |
| batch_expand_copy(TorchShapeRef batch_size) const | BatchTensorBase< Derived > | |
| batch_index(TorchSlice indices) const | BatchTensorBase< Derived > | |
| batch_index_put(TorchSlice indices, const torch::Tensor &other) | BatchTensorBase< Derived > | |
| batch_reshape(TorchShapeRef batch_shape) const | BatchTensorBase< Derived > | |
| batch_size(TorchSize index) const | BatchTensorBase< Derived > | |
| batch_sizes() const | BatchTensorBase< Derived > | |
| batch_sum(TorchSize d) const | BatchTensorBase< Derived > | |
| batch_transpose(TorchSize d1, TorchSize d2) const | BatchTensorBase< Derived > | |
| batch_unsqueeze(TorchSize d) const | BatchTensorBase< Derived > | |
| batched() const | BatchTensorBase< Derived > | |
| BatchTensorBase()=default | BatchTensorBase< Derived > | |
| BatchTensorBase(const torch::Tensor &tensor, TorchSize batch_dim) | BatchTensorBase< Derived > | |
| BatchTensorBase(const Derived &tensor) | BatchTensorBase< Derived > | |
| BatchTensorBase(Real)=delete | BatchTensorBase< Derived > | |
| clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) const | BatchTensorBase< Derived > | |
| const_base_dim | FixedDimTensor< Derived, S > | static |
| const_base_sizes | FixedDimTensor< Derived, S > | inlinestatic |
| const_base_storage | FixedDimTensor< Derived, S > | inlinestatic |
| cross(const VecBase< Derived2 > &v) const | VecBase< WR2 > | |
| detach() const | BatchTensorBase< Derived > | |
| dexp() const | WR2 | |
| dot(const VecBase< Derived2 > &v) const | VecBase< WR2 > | |
| drotate(const Rot &r) const | VecBase< WR2 > | |
| drotate(const R2 &R) const | VecBase< WR2 > | |
| empty(const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< Derived, S > | static |
| empty(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< Derived, S > | static |
| empty_like(const Derived &other) | BatchTensorBase< Derived > | static |
| exp() const | WR2 | |
| fill(const Real &v1, const Real &v2, const Real &v3, const torch::TensorOptions &options=default_tensor_options()) | VecBase< WR2 > | static |
| fill(const Scalar &v1, const Scalar &v2, const Scalar &v3) | VecBase< WR2 > | static |
| FixedDimTensor()=default | FixedDimTensor< Derived, S > | |
| FixedDimTensor(const torch::Tensor &tensor, TorchSize batch_dim) | FixedDimTensor< Derived, S > | explicit |
| FixedDimTensor(const torch::Tensor &tensor) | FixedDimTensor< Derived, S > | |
| full(Real init, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< Derived, S > | static |
| full(TorchShapeRef batch_shape, Real init, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< Derived, S > | static |
| full_like(const Derived &other, Real init) | BatchTensorBase< Derived > | static |
| identity_map(const torch::TensorOptions &options=default_tensor_options()) | VecBase< WR2 > | static |
| linspace(const Derived &start, const Derived &end, TorchSize nstep, TorchSize dim=0, TorchSize batch_dim=-1) | BatchTensorBase< Derived > | static |
| list_sum() const | BatchTensorBase< Derived > | |
| list_unsqueeze() const | BatchTensorBase< Derived > | |
| logspace(const Derived &start, const Derived &end, TorchSize nstep, TorchSize dim=0, TorchSize batch_dim=-1, Real base=10) | BatchTensorBase< Derived > | static |
| norm() const | VecBase< WR2 > | |
| norm_sq() const | VecBase< WR2 > | |
| ones(const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< Derived, S > | static |
| ones(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< Derived, S > | static |
| ones_like(const Derived &other) | BatchTensorBase< Derived > | static |
| operator BatchTensor() const | FixedDimTensor< Derived, S > | |
| operator()(TorchSize i, TorchSize j) const | WR2 | |
| VecBase< WR2 >::operator()(TorchSize i) const | VecBase< WR2 > | |
| operator-() const | BatchTensorBase< Derived > | |
| outer(const VecBase< Derived2 > &v) const | VecBase< WR2 > | |
| rotate(const Rot &r) const | VecBase< WR2 > | |
| rotate(const R2 &R) const | VecBase< WR2 > | |
| to(const torch::TensorOptions &options) const | BatchTensorBase< Derived > | |
| WR2(const R2 &T) | WR2 | |
| zeros(const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< Derived, S > | static |
| zeros(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< Derived, S > | static |
| zeros_like(const Derived &other) | BatchTensorBase< Derived > | static |
1.10.0