|
| Scalar (Real init, const TensorOptions &options) |
|
neml2::Tensor | base_unsqueeze_to (Size n) const |
| Expand base dimension to a given size.
|
|
| PrimitiveTensor ()=default |
| Special member functions.
|
|
| PrimitiveTensor (const ATensor &tensor, Size batch_dim) |
| Construct from another ATensor given batch dimension.
|
|
| PrimitiveTensor (const ATensor &tensor, const TraceableTensorShape &batch_shape) |
| Construct from another ATensor given batch shape.
|
|
| PrimitiveTensor (const Tensor &tensor) |
| Copy constructor.
|
|
| PrimitiveTensor (const ATensor &tensor) |
| Construct from another ATensor and infer batch dimension.
|
|
| operator Tensor () const |
| Implicit conversion to a Tensor and loses information on the fixed base shape.
|
|
| 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 (Real)=delete |
|
Scalar | variable_data () const |
|
Scalar | clone () const |
|
Scalar | detach () const |
| Discard function graph.
|
|
Scalar | to (const TensorOptions &options) const |
| Change tensor options.
|
|
Scalar | 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.
|
|
Scalar | 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.
|
|
Scalar | 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, Real v) |
|
void | base_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other) |
|
void | base_index_put_ (indexing::TensorIndicesRef indices, Real v) |
|
Scalar | batch_expand (const TraceableTensorShape &batch_shape) const |
|
Scalar | 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.
|
|
Scalar | 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.
|
|
Scalar | 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.
|
|
Scalar | batch_reshape (const TraceableTensorShape &batch_shape) const |
| Reshape batch dimensions.
|
|
neml2::Tensor | base_reshape (TensorShapeRef base_shape) const |
| Reshape base dimensions.
|
|
Scalar | batch_unsqueeze (Size d) const |
| Unsqueeze a batch dimension.
|
|
neml2::Tensor | base_unsqueeze (Size d) const |
| Unsqueeze a base dimension.
|
|
Scalar | 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.
|
|