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

This is the complete list of members for FillWR2, 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
doc() constNEML2Objectinline
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
expected_options()FillWR2static
factory() constNEML2Objectinline
fill(Args &&... args)PrimitiveTensor< WR2, 3 >static
FillWR2(const OptionSet &options)FillWR2
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
get_data(const std::string &name)NEML2Object
get_driver(const std::string &name)NEML2Object
get_model(const std::string &name)NEML2Object
get_object(const std::string &section, const std::string &name)NEML2Object
get_scheduler(const std::string &name)NEML2Object
get_solver(const std::string &name)NEML2Object
host() constNEML2Object
host()NEML2Object
input_options() constNEML2Objectinline
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 >
make() const overrideFillWR2protectedvirtual
name() constNEML2Objectinline
NEML2Object()=deleteNEML2Object
NEML2Object(NEML2Object &&)=deleteNEML2Object
NEML2Object(const NEML2Object &)=deleteNEML2Object
NEML2Object(const OptionSet &options)NEML2Object
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 >
operator=(NEML2Object &&)=deleteNEML2Object
operator=(const NEML2Object &)=deleteNEML2Object
path() constNEML2Objectinline
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
resolve_tensor(const std::string &name)NEML2Object
rotate(const Rot &r) constVecBase< WR2 >
rotate(const R2 &R) constVecBase< WR2 >
settings() constNEML2Objectinline
setup() overrideUserTensorBase< WR2 >virtual
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 >
tensor_type()UserTensorBase< WR2 >protectedstatic
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 >
type() constNEML2Objectinline
UserTensorBase(const OptionSet &options)UserTensorBase< 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
~NEML2Object()=defaultNEML2Objectvirtual