template<bool scaled>
struct neml2::NonlinearSystem::Res< scaled >
Convenience struct to hold residual to prevent developers from accidentally confusing the scaled and unscaled residual
|
| | Res ()=default |
| |
| | Res (const Tensor &r) |
| | Conversion from Tensor must be explicit.
|
| |
| | Res (const Res<!scaled > &r) |
| | Conversion between scaled and unscaled must be explicit.
|
| |
| | Tensor ()=default |
| | Special member functions.
|
| |
| | Tensor (const ATensor &tensor, Size batch_dim) |
| | Construct from another ATensor.
|
| |
| | Tensor (const ATensor &tensor, const TraceableTensorShape &batch_shape) |
| | Construct from another ATensor with given batch shape.
|
| |
| template<class Derived > |
| | Tensor (const TensorBase< Derived > &tensor) |
| | Copy from TensorBase.
|
| |
| Tensor | base_unsqueeze_to (Size n) const |
| | Expand base dimension to a given size.
|
| |
| | 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 |
| |
| Tensor | variable_data () const |
| |
| Tensor | clone () const |
| |
| Tensor | detach () const |
| | Discard function graph.
|
| |
| Tensor | to (const TensorOptions &options) const |
| | Change tensor options.
|
| |
| Tensor | 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.
|
| |
| Tensor | 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.
|
| |
| Tensor | 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) |
| |
| Tensor | batch_expand (const TraceableTensorShape &batch_shape) const |
| |
| Tensor | 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.
|
| |
| Tensor | 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.
|
| |
| Tensor | 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.
|
| |
| Tensor | batch_reshape (const TraceableTensorShape &batch_shape) const |
| | Reshape batch dimensions.
|
| |
| neml2::Tensor | base_reshape (TensorShapeRef base_shape) const |
| | Reshape base dimensions.
|
| |
| Tensor | batch_unsqueeze (Size d) const |
| | Unsqueeze a batch dimension.
|
| |
| neml2::Tensor | base_unsqueeze (Size d) const |
| | Unsqueeze a base dimension.
|
| |
| Tensor | 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.
|
| |
|
| static Tensor | create (const TensorDataContainer &data, const TensorOptions &options=default_tensor_options()) |
| | Arbitrary (unbatched) tensor from a nested container.
|
| |
| static Tensor | create (const TensorDataContainer &data, Size batch_dim, const TensorOptions &options=default_tensor_options()) |
| | Arbitrary tensor from a nested container.
|
| |
| static Tensor | empty (TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) |
| | Unbatched empty tensor given base shape.
|
| |
| static Tensor | empty (const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) |
| | Empty tensor given batch and base shapes.
|
| |
| static Tensor | zeros (TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) |
| | Unbatched tensor filled with zeros given base shape.
|
| |
| static Tensor | zeros (const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) |
| | Zero tensor given batch and base shapes.
|
| |
| static Tensor | ones (TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) |
| | Unbatched tensor filled with ones given base shape.
|
| |
| static Tensor | ones (const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) |
| | Unit tensor given batch and base shapes.
|
| |
| static Tensor | full (TensorShapeRef base_shape, const CScalar &init, const TensorOptions &options=default_tensor_options()) |
| | Unbatched tensor filled with a given value given base shape.
|
| |
| static Tensor | full (const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const CScalar &init, const TensorOptions &options=default_tensor_options()) |
| | Full tensor given batch and base shapes.
|
| |
| static Tensor | identity (Size n, const TensorOptions &options=default_tensor_options()) |
| | Unbatched identity tensor.
|
| |
| static Tensor | identity (const TraceableTensorShape &batch_shape, Size n, const TensorOptions &options=default_tensor_options()) |
| | Identity tensor given batch shape and base length.
|
| |
| static Tensor | empty_like (const Tensor &other) |
| |
| static Tensor | zeros_like (const Tensor &other) |
| | Zero tensor like another, i.e. same batch and base shapes, same tensor options, etc.
|
| |
| static Tensor | ones_like (const Tensor &other) |
| | Unit tensor like another, i.e. same batch and base shapes, same tensor options, etc.
|
| |
| static Tensor | full_like (const Tensor &other, const CScalar &init) |
| |
| static Tensor | linspace (const Tensor &start, const Tensor &end, Size nstep, Size dim=0) |
| | Create a new tensor by adding a new batch dimension with linear spacing between start and end.
|
| |
| static Tensor | logspace (const Tensor &start, const Tensor &end, Size nstep, Size dim=0, const CScalar &base=10) |
| | log-space equivalent of the linspace named constructor
|
| |
| TraceableTensorShape | _batch_sizes |
| | Traceable batch sizes.
|
| |