NEML2 2.0.0
Loading...
Searching...
No Matches
Vec Member List

This is the complete list of members for Vec, including all inherited members.

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