NEML2 2.0.0
Loading...
Searching...
No Matches
PrimitiveTensor< Derived, S > Member List

This is the complete list of members for PrimitiveTensor< Derived, S >, including all inherited members.

base_dim() constTensorBase< Derived >
base_expand(TensorShapeRef shape) constTensorBase< Derived >
base_expand(Size size, Size d) constTensorBase< Derived >
base_expand_as(const neml2::Tensor &other) constTensorBase< Derived >
base_flatten() constTensorBase< Derived >
base_index(indexing::TensorIndicesRef indices) constTensorBase< Derived >
base_index_put_(indexing::TensorIndicesRef indices, const ATensor &other)TensorBase< Derived >
base_index_put_(indexing::TensorIndicesRef indices, const CScalar &v)TensorBase< Derived >
base_movedim(Size old_dim, Size new_dim) constTensorBase< Derived >
base_reshape(TensorShapeRef shape) constTensorBase< Derived >
base_size(Size i) constTensorBase< Derived >
base_sizes() constTensorBase< Derived >
base_sizes_sequence typedefPrimitiveTensor< Derived, S >
base_slice(Size d, const indexing::Slice &index) constTensorBase< Derived >
base_squeeze(Size d) constTensorBase< Derived >
base_transpose(Size d1, Size d2) constTensorBase< Derived >
base_unsqueeze(Size d, Size n=1) constTensorBase< Derived >
batch_dim() constTensorBase< Derived >
batch_expand(const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) constTensorBase< Derived >
batch_expand_as(const neml2::Tensor &other) constTensorBase< Derived >
batch_flatten() constTensorBase< Derived >
batch_reshape(const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape) constTensorBase< Derived >
batch_size(Size i) constTensorBase< Derived >
batch_sizes() constTensorBase< Derived >
clone() constTensorBase< Derived >
const_base_dimPrimitiveTensor< Derived, S >static
const_base_numelPrimitiveTensor< Derived, S >static
const_base_sizesPrimitiveTensor< Derived, S >inlinestatic
contiguous() constTensorBase< Derived >
create(const TensorDataContainer &data, Size intmd_dim=0, const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
detach() constTensorBase< Derived >
dynamic_dim() constTensorBase< Derived >
dynamic_expand(const TraceableTensorShape &shape) constTensorBase< Derived >
dynamic_expand(const TraceableSize &size, Size d) constTensorBase< Derived >
dynamic_expand_as(const neml2::Tensor &other) constTensorBase< Derived >
dynamic_flatten() constTensorBase< Derived >
dynamic_index(indexing::TensorIndicesRef indices) constTensorBase< Derived >
dynamic_index_put_(indexing::TensorIndicesRef indices, const ATensor &other)TensorBase< Derived >
dynamic_index_put_(indexing::TensorIndicesRef indices, const CScalar &v)TensorBase< Derived >
dynamic_movedim(Size old_dim, Size new_dim) constTensorBase< Derived >
dynamic_reshape(const TraceableTensorShape &shape) constTensorBase< Derived >
dynamic_size(Size i) constTensorBase< Derived >
dynamic_sizes() constTensorBase< Derived >
dynamic_slice(Size d, const indexing::Slice &index) constTensorBase< Derived >
dynamic_squeeze(Size d) constTensorBase< Derived >
dynamic_transpose(Size d1, Size d2) constTensorBase< Derived >
dynamic_unsqueeze(Size d, Size n=1) constTensorBase< Derived >
einsum(c10::string_view equation, TensorList tensors)PrimitiveTensor< Derived, S >static
empty(const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
empty(const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
empty_like(const Derived &other)TensorBase< Derived >static
fill(Args &&... args)PrimitiveTensor< Derived, S >static
full(const CScalar &init, const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
full(const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, const CScalar &init, const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
full_like(const Derived &other, const CScalar &init)TensorBase< Derived >static
intmd_dim() constTensorBase< Derived >
intmd_expand(TensorShapeRef shape) constTensorBase< Derived >
intmd_expand(Size size, Size d) constTensorBase< Derived >
intmd_expand_as(const neml2::Tensor &other) constTensorBase< Derived >
intmd_flatten() constTensorBase< Derived >
intmd_index(indexing::TensorIndicesRef indices) constTensorBase< Derived >
intmd_index_put_(indexing::TensorIndicesRef indices, const ATensor &other)TensorBase< Derived >
intmd_index_put_(indexing::TensorIndicesRef indices, const CScalar &v)TensorBase< Derived >
intmd_movedim(Size old_dim, Size new_dim) constTensorBase< Derived >
intmd_reshape(TensorShapeRef shape) constTensorBase< Derived >
intmd_size(Size i) constTensorBase< Derived >
intmd_sizes() constTensorBase< Derived >
intmd_slice(Size d, const indexing::Slice &index) constTensorBase< Derived >
intmd_squeeze(Size d) constTensorBase< Derived >
intmd_transpose(Size d1, Size d2) constTensorBase< Derived >
intmd_unsqueeze(Size d, Size n=1) constTensorBase< Derived >
ones(const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
ones(const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
ones_like(const Derived &other)TensorBase< Derived >static
operator neml2::Tensor() constPrimitiveTensor< Derived, S >
operator()(Args... i) constPrimitiveTensor< Derived, S >
operator-() constTensorBase< Derived >
PrimitiveTensor()=defaultPrimitiveTensor< Derived, S >
PrimitiveTensor(const ATensor &tensor, Size intmd_dim)PrimitiveTensor< Derived, S >
PrimitiveTensor(const ATensor &tensor, Size dynamic_dim, Size intmd_dim)PrimitiveTensor< Derived, S >
PrimitiveTensor(const ATensor &tensor, const TraceableTensorShape &dynamic_shape, Size intmd_dim)PrimitiveTensor< Derived, S >
PrimitiveTensor(const TensorBase< Derived2 > &tensor)PrimitiveTensor< Derived, S >
rand(const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
rand(const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape, const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
rand_like(const Derived &other)TensorBase< Derived >static
static_dim() constTensorBase< Derived >
static_expand(TensorShapeRef intmd_shape, TensorShapeRef base_shape) constTensorBase< Derived >
static_expand_as(const neml2::Tensor &other) constTensorBase< Derived >
static_flatten() constTensorBase< Derived >
static_reshape(TensorShapeRef intmd_shape, TensorShapeRef base_shape) constTensorBase< Derived >
static_size(Size i) constTensorBase< Derived >
static_sizes() constTensorBase< Derived >
TensorBase()=defaultTensorBase< Derived >
TensorBase(const ATensor &tensor, Size dynamic_dim, Size intmd_dim)TensorBase< Derived >
TensorBase(const ATensor &tensor, TraceableTensorShape dynamic_shape, Size intmd_dim)TensorBase< Derived >
TensorBase(const TensorBase< Derived2 > &tensor)TensorBase< Derived >inline
TensorBase(double)=deleteTensorBase< Derived >
TensorBase(float)=deleteTensorBase< Derived >
TensorBase(int)=deleteTensorBase< Derived >
to(const TensorOptions &options) constTensorBase< Derived >
validate_shapes_and_dims() constPrimitiveTensor< Derived, S >protected
variable_data() constTensorBase< Derived >
zeros(const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
zeros(const TraceableTensorShape &dynamic_shape, TensorShapeRef intmd_shape={}, const TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
zeros_like(const Derived &other)TensorBase< Derived >static