#include <Tensor.h>

Inheritance diagram for Tensor:

[legend]

Public Member Functions
	Tensor ()=default
	Special member functions.

	Tensor (const ATensor &tensor, Size dynamic_dim, Size intmd_dim=0)
	Construct from another ATensor with inferred dynamic shape.

	Tensor (const ATensor &tensor, const TraceableTensorShape &dynamic_shape, Size intmd_dim=0)
	Construct from another ATensor with given dynamic shape.

template<class Derived >
	Tensor (const TensorBase< Derived > &tensor)
	Copy from TensorBase.

Public Member Functions inherited from TensorBase< Tensor >
	TensorBase ()=default
	Default constructor.

	TensorBase (const ATensor &tensor, Size dynamic_dim, Size intmd_dim)
	Construct from an ATensor with given dynamic dimension.

	TensorBase (const ATensor &tensor, TraceableTensorShape dynamic_shape, Size intmd_dim)
	Construct from an ATensor with given dynamic shape.

	TensorBase (const TensorBase< Derived2 > &tensor)
	Copy constructor.

	TensorBase (double)=delete

	TensorBase (float)=delete

	TensorBase (int)=delete

Tensor	variable_data () const
	Variable data without function graph.

Tensor	contiguous () const

Tensor	clone () const
	Clone (take ownership)

Tensor	detach () const
	Discard function graph.

Tensor	to (const TensorOptions &options) const
	Change tensor options.

Tensor	operator- () const
	Negation.

Size	batch_dim () const

Size	base_dim () const

Size	dynamic_dim () const

Size	static_dim () const

Size	intmd_dim () const

TraceableTensorShape	batch_sizes () const

TensorShapeRef	base_sizes () const

const TraceableTensorShape &	dynamic_sizes () const

TensorShapeRef	static_sizes () const

TensorShapeRef	intmd_sizes () const

TraceableSize	batch_size (Size i) const

Size	base_size (Size i) const

const TraceableSize &	dynamic_size (Size i) const

Size	static_size (Size i) const

Size	intmd_size (Size i) const

Tensor	dynamic_index (indexing::TensorIndicesRef indices) const

Tensor	intmd_index (indexing::TensorIndicesRef indices) const

neml2::Tensor	base_index (indexing::TensorIndicesRef indices) const

Tensor	dynamic_slice (Size d, const indexing::Slice &index) const

Tensor	intmd_slice (Size d, const indexing::Slice &index) const

neml2::Tensor	base_slice (Size d, const indexing::Slice &index) const

void	dynamic_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other)

void	dynamic_index_put_ (indexing::TensorIndicesRef indices, const CScalar &v)

void	intmd_index_put_ (indexing::TensorIndicesRef indices, const ATensor &other)

void	intmd_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	dynamic_expand (const TraceableTensorShape &shape) const

Tensor	intmd_expand (TensorShapeRef shape) const

neml2::Tensor	base_expand (TensorShapeRef shape) const

Tensor	batch_expand (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) const

neml2::Tensor	static_expand (TensorShapeRef intmd_shape, TensorShapeRef base_shape) const

Tensor	dynamic_expand (const TraceableSize &size, Size d) const

Tensor	intmd_expand (Size size, Size d) const

neml2::Tensor	base_expand (Size size, Size d) const

Tensor	dynamic_expand_as (const neml2::Tensor &other) const

Tensor	intmd_expand_as (const neml2::Tensor &other) const

neml2::Tensor	base_expand_as (const neml2::Tensor &other) const

Tensor	batch_expand_as (const neml2::Tensor &other) const

neml2::Tensor	static_expand_as (const neml2::Tensor &other) const

Tensor	dynamic_reshape (const TraceableTensorShape &shape) const

Tensor	intmd_reshape (TensorShapeRef shape) const

neml2::Tensor	base_reshape (TensorShapeRef shape) const

Tensor	batch_reshape (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) const

neml2::Tensor	static_reshape (TensorShapeRef intmd_shape, TensorShapeRef base_shape) const

Tensor	dynamic_squeeze (Size d) const

Tensor	intmd_squeeze (Size d) const

neml2::Tensor	base_squeeze (Size d) const

Tensor	dynamic_unsqueeze (Size d, Size n=1) const

Tensor	intmd_unsqueeze (Size d, Size n=1) const

neml2::Tensor	base_unsqueeze (Size d, Size n=1) const

Tensor	dynamic_transpose (Size d1, Size d2) const

Tensor	intmd_transpose (Size d1, Size d2) const

neml2::Tensor	base_transpose (Size d1, Size d2) const

Tensor	dynamic_movedim (Size old_dim, Size new_dim) const

Tensor	intmd_movedim (Size old_dim, Size new_dim) const

neml2::Tensor	base_movedim (Size old_dim, Size new_dim) const

Tensor	dynamic_flatten () const

Tensor	intmd_flatten () const

neml2::Tensor	base_flatten () const

Tensor	batch_flatten () const
	Flatten batch dimensions.

neml2::Tensor	static_flatten () const
	Flatten static dimensions.

Static Public Member Functions
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 dynamic_dim, Size intmd_dim=0, const TensorOptions &options=default_tensor_options())
	Arbitrary tensor from a nested container.

static Tensor	identity (Size n, const TensorOptions &options=default_tensor_options())
	Identity tensor.


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

static Tensor	empty (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options())


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

static Tensor	zeros (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options())


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

static Tensor	ones (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options())


static Tensor	full (TensorShapeRef base_shape, const CScalar &init, const TensorOptions &options=default_tensor_options())

static Tensor	full (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, TensorShapeRef base_shape, const CScalar &init, const TensorOptions &options=default_tensor_options())


static Tensor	rand (TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options())

static Tensor	rand (const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options())

Static Public Member Functions inherited from TensorBase< Tensor >
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	rand_like (const Tensor &other)

Additional Inherited Members
Protected Member Functions inherited from TensorBase< Tensor >
void	validate_shapes_and_dims () const
	Validate shapes and dimensions.

Constructor & Destructor Documentation

◆ Tensor() [1/4]

Tensor ( )

default

Special member functions.

◆ Tensor() [2/4]

Tensor	(	const ATensor &	tensor,
		Size	dynamic_dim,
		Size	intmd_dim = 0 )

Construct from another ATensor with inferred dynamic shape.

◆ Tensor() [3/4]

Tensor	(	const ATensor &	tensor,
		const TraceableTensorShape &	dynamic_shape,
		Size	intmd_dim = 0 )

Construct from another ATensor with given dynamic shape.

◆ Tensor() [4/4]

template<class Derived >

Tensor ( const TensorBase< Derived > & tensor )

inline

Copy from TensorBase.

Member Function Documentation

◆ create() [1/2]

Tensor create	(	const TensorDataContainer &	data,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

Arbitrary (unbatched) tensor from a nested container.

◆ create() [2/2]

Tensor create	(	const TensorDataContainer &	data,
		Size	dynamic_dim,
		Size	intmd_dim = 0,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

Arbitrary tensor from a nested container.

◆ empty() [1/2]

Tensor empty	(	const TraceableTensorShape &	dynamic_shape,
		TensorShapeRef	intmd_shape,
		TensorShapeRef	base_shape,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

◆ empty() [2/2]

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

staticnodiscard

Empty tensor filled with undefined values

◆ full() [1/2]

Tensor full	(	const TraceableTensorShape &	dynamic_shape,
		TensorShapeRef	intmd_shape,
		TensorShapeRef	base_shape,
		const CScalar &	init,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

◆ full() [2/2]

Tensor full	(	TensorShapeRef	base_shape,
		const CScalar &	init,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

Tensor filled with a given value

◆ identity()

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

staticnodiscard

Identity tensor.

◆ ones() [1/2]

Tensor ones	(	const TraceableTensorShape &	dynamic_shape,
		TensorShapeRef	intmd_shape,
		TensorShapeRef	base_shape,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

◆ ones() [2/2]

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

staticnodiscard

Tensor filled with ones

◆ rand() [1/2]

Tensor rand	(	const TraceableTensorShape &	dynamic_shape,
		TensorShapeRef	intmd_shape,
		TensorShapeRef	base_shape,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

◆ rand() [2/2]

Tensor rand	(	TensorShapeRef	base_shape,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

Tensor filled with random values from a uniform distribution [0, 1)

◆ zeros() [1/2]

Tensor zeros	(	const TraceableTensorShape &	dynamic_shape,
		TensorShapeRef	intmd_shape,
		TensorShapeRef	base_shape,
		const TensorOptions &	options = default_tensor_options() )

staticnodiscard

◆ zeros() [2/2]

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

staticnodiscard

Tensor filled with zeros

Public Member Functions

Static Public Member Functions

Additional Inherited Members

Constructor & Destructor Documentation

◆ Tensor() [1/4]

◆ Tensor() [2/4]

◆ Tensor() [3/4]

◆ Tensor() [4/4]

Member Function Documentation

◆ create() [1/2]

◆ create() [2/2]

◆ empty() [1/2]

◆ empty() [2/2]

◆ full() [1/2]

◆ full() [2/2]

◆ identity()

◆ ones() [1/2]

◆ ones() [2/2]

◆ rand() [1/2]

◆ rand() [2/2]

◆ zeros() [1/2]

◆ zeros() [2/2]