Tensor Class Reference

#include <Tensor.h>

Inheritance diagram for Tensor:

Static Public Member Functions
static Tensor	empty (const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())
	Unbatched empty tensor given base shape.
static Tensor	empty (const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())
	Empty tensor given batch and base shapes.
static Tensor	zeros (const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())
	Unbatched tensor filled with zeros given base shape.
static Tensor	zeros (const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())
	Zero tensor given batch and base shapes.
static Tensor	ones (const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())
	Unbatched tensor filled with ones given base shape.
static Tensor	ones (const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())
	Unit tensor given batch and base shapes.
static Tensor	full (const TensorShapeRef &base_shape, Real init, const torch::TensorOptions &options=default_tensor_options())
	Unbatched tensor filled with a given value given base shape.
static Tensor	full (const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, Real init, const torch::TensorOptions &options=default_tensor_options())
	Full tensor given batch and base shapes.
static Tensor	identity (Size n, const torch::TensorOptions &options=default_tensor_options())
	Unbatched identity tensor.
static Tensor	identity (const TensorShapeRef &batch_shape, Size n, const torch::TensorOptions &options=default_tensor_options())
	Identity tensor given batch shape and base length.
Static Public Member Functions inherited from TensorBase< Tensor >
static Tensor	empty_like (const Tensor &other)
	Empty tensor like another, i.e. same batch and base shapes, same tensor options, etc.
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, Real init)
static Tensor	linspace (const Tensor &start, const Tensor &end, Size nstep, Size dim=0, Size batch_dim=-1)
	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, Size batch_dim=-1, Real base=10)
	log-space equivalent of the linspace named constructor

Additional Inherited Members
Public Member Functions inherited from TensorBase< Tensor >
	TensorBase ()=default
	Default constructor.
	TensorBase (const torch::Tensor &tensor, Size batch_dim)
	Construct from another torch::Tensor.
	TensorBase (const Tensor &tensor)
	Copy constructor.
	TensorBase (Real)=delete
Tensor	clone (torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) const
Tensor	detach () const
	Discard function graph.
Tensor	to (const torch::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.
TensorShapeRef	batch_sizes () const
	Return the batch size.
Size	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.
void	batch_index_put_ (indexing::TensorIndicesRef indices, const torch::Tensor &other)
	Set values by slicing on the batch dimensions.
void	base_index_put_ (indexing::TensorIndicesRef indices, const torch::Tensor &other)
	Set values by slicing on the base dimensions.
Tensor	batch_expand (TensorShapeRef batch_size) const
neml2::Tensor	base_expand (TensorShapeRef base_size) const
	Return a new view of the tensor with values broadcast along the base dimensions.
Tensor	batch_expand_as (const Derived2 &other) const
	Expand the batch to have the same shape as another tensor.
Derived2	base_expand_as (const Derived2 &other) const
	Expand the base to have the same shape as another tensor.
Tensor	batch_expand_copy (TensorShapeRef batch_size) const
	Return a new tensor with values broadcast along the batch dimensions.
neml2::Tensor	base_expand_copy (TensorShapeRef base_size) const
	Return a new tensor with values broadcast along the base dimensions.
Tensor	batch_reshape (TensorShapeRef 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.

Member Function Documentation

empty() [1/2]

Tensor empty ( const TensorShapeRef & base_shape, const torch::TensorOptions & options = default_tensor_options() )

static

Unbatched empty tensor given base shape.

empty() [2/2]

Tensor empty ( const TensorShapeRef & batch_shape, const TensorShapeRef & base_shape, const torch::TensorOptions & options = default_tensor_options() )

static

Empty tensor given batch and base shapes.

full() [1/2]

Tensor full ( const TensorShapeRef & base_shape, Real init, const torch::TensorOptions & options = default_tensor_options() )

static

Unbatched tensor filled with a given value given base shape.

full() [2/2]

Tensor full ( const TensorShapeRef & batch_shape, const TensorShapeRef & base_shape, Real init, const torch::TensorOptions & options = default_tensor_options() )

static

Full tensor given batch and base shapes.

identity() [1/2]

Tensor identity ( const TensorShapeRef & batch_shape, Size n, const torch::TensorOptions & options = default_tensor_options() )

static

Identity tensor given batch shape and base length.

identity() [2/2]

Tensor identity ( Size n, const torch::TensorOptions & options = default_tensor_options() )

static

Unbatched identity tensor.

ones() [1/2]

Tensor ones ( const TensorShapeRef & base_shape, const torch::TensorOptions & options = default_tensor_options() )

static

Unbatched tensor filled with ones given base shape.

ones() [2/2]

Tensor ones ( const TensorShapeRef & batch_shape, const TensorShapeRef & base_shape, const torch::TensorOptions & options = default_tensor_options() )

static

Unit tensor given batch and base shapes.

zeros() [1/2]

Tensor zeros ( const TensorShapeRef & base_shape, const torch::TensorOptions & options = default_tensor_options() )

static

Unbatched tensor filled with zeros given base shape.

zeros() [2/2]

Tensor zeros ( const TensorShapeRef & batch_shape, const TensorShapeRef & base_shape, const torch::TensorOptions & options = default_tensor_options() )

static

Zero tensor given batch and base shapes.