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

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

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