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

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

base_dim() constBatchTensorBase< Derived >
base_expand(TorchShapeRef base_size) constBatchTensorBase< Derived >
base_expand_as(const Derived2 &other) constBatchTensorBase< Derived >
base_expand_copy(TorchShapeRef base_size) constBatchTensorBase< Derived >
base_index(const TorchSlice &indices) constBatchTensorBase< Derived >
base_index_put(const TorchSlice &indices, const torch::Tensor &other)BatchTensorBase< Derived >
base_movedim(TorchSize d1, TorchSize d2) constBatchTensorBase< Derived >
base_reshape(TorchShapeRef base_shape) constBatchTensorBase< Derived >
base_size(TorchSize index) constBatchTensorBase< Derived >
base_sizes() constBatchTensorBase< Derived >
base_storage() constBatchTensorBase< Derived >
base_transpose(TorchSize d1, TorchSize d2) constBatchTensorBase< Derived >
base_unsqueeze(TorchSize d) constBatchTensorBase< Derived >
batch_dim() constBatchTensorBase< Derived >
batch_dim()BatchTensorBase< Derived >
batch_expand(TorchShapeRef batch_size) constBatchTensorBase< Derived >
batch_expand_as(const Derived2 &other) constBatchTensorBase< Derived >
batch_expand_copy(TorchShapeRef batch_size) constBatchTensorBase< Derived >
batch_index(TorchSlice indices) constBatchTensorBase< Derived >
batch_index_put(TorchSlice indices, const torch::Tensor &other)BatchTensorBase< Derived >
batch_reshape(TorchShapeRef batch_shape) constBatchTensorBase< Derived >
batch_size(TorchSize index) constBatchTensorBase< Derived >
batch_sizes() constBatchTensorBase< Derived >
batch_sum(TorchSize d) constBatchTensorBase< Derived >
batch_transpose(TorchSize d1, TorchSize d2) constBatchTensorBase< Derived >
batch_unsqueeze(TorchSize d) constBatchTensorBase< Derived >
batched() constBatchTensorBase< Derived >
BatchTensorBase()=defaultBatchTensorBase< Derived >
BatchTensorBase(const torch::Tensor &tensor, TorchSize batch_dim)BatchTensorBase< Derived >
BatchTensorBase(const Derived &tensor)BatchTensorBase< Derived >
BatchTensorBase(Real)=deleteBatchTensorBase< Derived >
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) constBatchTensorBase< Derived >
const_base_dimFixedDimTensor< Derived, S >static
const_base_sizesFixedDimTensor< Derived, S >inlinestatic
const_base_storageFixedDimTensor< Derived, S >inlinestatic
cross(const VecBase< Derived2 > &v) constVecBase< WR2 >
detach() constBatchTensorBase< Derived >
dexp() constWR2
doc() constNEML2Objectinline
dot(const VecBase< Derived2 > &v) constVecBase< WR2 >
drotate(const Rot &r) constVecBase< WR2 >
drotate(const R2 &R) constVecBase< WR2 >
empty(const torch::TensorOptions &options=default_tensor_options())FixedDimTensor< Derived, S >static
empty(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options())FixedDimTensor< Derived, S >static
empty_like(const Derived &other)BatchTensorBase< Derived >static
exp() constWR2
expected_options()FillWR2static
neml2::WR2::fill(const Real &v1, const Real &v2, const Real &v3, const torch::TensorOptions &options=default_tensor_options())VecBase< WR2 >static
neml2::WR2::fill(const Scalar &v1, const Scalar &v2, const Scalar &v3)VecBase< WR2 >static
FillWR2(const OptionSet &options)FillWR2
FixedDimTensor()=defaultFixedDimTensor< Derived, S >
FixedDimTensor(const torch::Tensor &tensor, TorchSize batch_dim)FixedDimTensor< Derived, S >explicit
FixedDimTensor(const torch::Tensor &tensor)FixedDimTensor< Derived, S >
full(Real init, const torch::TensorOptions &options=default_tensor_options())FixedDimTensor< Derived, S >static
full(TorchShapeRef batch_shape, Real init, const torch::TensorOptions &options=default_tensor_options())FixedDimTensor< Derived, S >static
full_like(const Derived &other, Real init)BatchTensorBase< Derived >static
host() constNEML2Object
host()NEML2Object
identity_map(const torch::TensorOptions &options=default_tensor_options())VecBase< WR2 >static
input_options() constNEML2Objectinline
linspace(const Derived &start, const Derived &end, TorchSize nstep, TorchSize dim=0, TorchSize batch_dim=-1)BatchTensorBase< Derived >static
list_sum() constBatchTensorBase< Derived >
list_unsqueeze() constBatchTensorBase< Derived >
logspace(const Derived &start, const Derived &end, TorchSize nstep, TorchSize dim=0, TorchSize batch_dim=-1, Real base=10)BatchTensorBase< Derived >static
name() constNEML2Objectinline
NEML2Object(const OptionSet &options)NEML2Object
norm() constVecBase< WR2 >
norm_sq() constVecBase< WR2 >
ones(const torch::TensorOptions &options=default_tensor_options())FixedDimTensor< Derived, S >static
ones(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options())FixedDimTensor< Derived, S >static
ones_like(const Derived &other)BatchTensorBase< Derived >static
operator BatchTensor() constFixedDimTensor< Derived, S >
operator()(TorchSize i, TorchSize j) constWR2
VecBase< WR2 >::operator()(TorchSize i) constVecBase< WR2 >
operator-() constBatchTensorBase< Derived >
outer(const VecBase< Derived2 > &v) constVecBase< WR2 >
path() constNEML2Objectinline
rotate(const Rot &r) constVecBase< WR2 >
rotate(const R2 &R) constVecBase< WR2 >
setup()NEML2Objectinlinevirtual
to(const torch::TensorOptions &options) constBatchTensorBase< Derived >
type() constNEML2Objectinline
UserTensor(const OptionSet &options)UserTensor
WR2(const R2 &T)WR2
zeros(const torch::TensorOptions &options=default_tensor_options())FixedDimTensor< Derived, S >static
zeros(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options())FixedDimTensor< Derived, S >static
zeros_like(const Derived &other)BatchTensorBase< Derived >static
~NEML2Object()=defaultNEML2Objectvirtual