|
NEML2 2.0.0
|
This is the complete list of members for Tensor, including all inherited members.
| _batch_sizes | TensorBase< Tensor > | protected |
| base_dim() const | TensorBase< Tensor > | |
| base_expand(TensorShapeRef base_shape) const | TensorBase< Tensor > | |
| base_expand(Size base_size, Size dim) const | TensorBase< Tensor > | |
| base_expand_as(const neml2::Tensor &other) const | TensorBase< Tensor > | |
| base_expand_copy(TensorShapeRef base_shape) const | TensorBase< Tensor > | |
| base_flatten() const | TensorBase< Tensor > | |
| base_index(indexing::TensorIndicesRef indices) const | TensorBase< Tensor > | |
| base_index_put_(indexing::TensorIndicesRef indices, const ATensor &other) | TensorBase< Tensor > | |
| base_index_put_(indexing::TensorIndicesRef indices, const CScalar &v) | TensorBase< Tensor > | |
| base_reshape(TensorShapeRef base_shape) const | TensorBase< Tensor > | |
| base_size(Size index) const | TensorBase< Tensor > | |
| base_sizes() const | TensorBase< Tensor > | |
| base_slice(Size dim, const indexing::Slice &index) const | TensorBase< Tensor > | |
| base_storage() const | TensorBase< Tensor > | |
| base_transpose(Size d1, Size d2) const | TensorBase< Tensor > | |
| base_unsqueeze(Size d) const | TensorBase< Tensor > | |
| base_unsqueeze_to(Size n) const | Tensor | |
| batch_dim() const | TensorBase< Tensor > | |
| batch_expand(const TraceableTensorShape &batch_shape) const | TensorBase< Tensor > | |
| batch_expand(const TraceableSize &batch_size, Size dim) const | TensorBase< Tensor > | |
| batch_expand_as(const neml2::Tensor &other) const | TensorBase< Tensor > | |
| batch_expand_copy(const TraceableTensorShape &batch_shape) const | TensorBase< Tensor > | |
| batch_index(indexing::TensorIndicesRef indices) const | TensorBase< Tensor > | |
| batch_index_put_(indexing::TensorIndicesRef indices, const ATensor &other) | TensorBase< Tensor > | |
| batch_index_put_(indexing::TensorIndicesRef indices, const CScalar &v) | TensorBase< Tensor > | |
| batch_reshape(const TraceableTensorShape &batch_shape) const | TensorBase< Tensor > | |
| batch_size(Size index) const | TensorBase< Tensor > | |
| batch_sizes() const | TensorBase< Tensor > | |
| batch_slice(Size dim, const indexing::Slice &index) const | TensorBase< Tensor > | |
| batch_transpose(Size d1, Size d2) const | TensorBase< Tensor > | |
| batch_unsqueeze(Size d) const | TensorBase< Tensor > | |
| batched() const | TensorBase< Tensor > | |
| clone() const | TensorBase< Tensor > | |
| create(const TensorDataContainer &data, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| create(const TensorDataContainer &data, Size batch_dim, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| detach() const | TensorBase< Tensor > | |
| empty(TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| empty(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| empty_like(const Tensor &other) | TensorBase< Tensor > | static |
| full(TensorShapeRef base_shape, const CScalar &init, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| full(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const CScalar &init, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| full_like(const Tensor &other, const CScalar &init) | TensorBase< Tensor > | static |
| identity(Size n, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| identity(const TraceableTensorShape &batch_shape, Size n, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| linspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0) | TensorBase< Tensor > | static |
| logspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0, const CScalar &base=10) | TensorBase< Tensor > | static |
| ones(TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| ones(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| ones_like(const Tensor &other) | TensorBase< Tensor > | static |
| operator-() const | TensorBase< Tensor > | |
| Tensor()=default | Tensor | |
| Tensor(const ATensor &tensor, Size batch_dim) | Tensor | |
| Tensor(const ATensor &tensor, const TraceableTensorShape &batch_shape) | Tensor | |
| Tensor(const TensorBase< Derived > &tensor) | Tensor | inline |
| TensorBase()=default | TensorBase< Tensor > | |
| TensorBase(const ATensor &tensor, Size batch_dim) | TensorBase< Tensor > | |
| TensorBase(const ATensor &tensor, const TraceableTensorShape &batch_shape) | TensorBase< Tensor > | |
| TensorBase(const neml2::Tensor &tensor) | TensorBase< Tensor > | |
| TensorBase(double)=delete | TensorBase< Tensor > | |
| TensorBase(float)=delete | TensorBase< Tensor > | |
| TensorBase(int)=delete | TensorBase< Tensor > | |
| to(const TensorOptions &options) const | TensorBase< Tensor > | |
| variable_data() const | TensorBase< Tensor > | |
| zeros(TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| zeros(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| zeros_like(const Tensor &other) | TensorBase< Tensor > | static |