NEML2 1.4.0
|
This is the complete list of members for FillR2, including all inherited members.
base_dim() const | TensorBase< Derived > | |
base_expand(TensorShapeRef base_size) const | TensorBase< Derived > | |
base_expand_as(const Derived2 &other) const | TensorBase< Derived > | |
base_expand_copy(TensorShapeRef base_size) const | TensorBase< Derived > | |
base_index(indexing::TensorIndicesRef indices) const | TensorBase< Derived > | |
base_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other) | TensorBase< Derived > | |
base_reshape(TensorShapeRef base_shape) const | TensorBase< Derived > | |
base_size(Size index) const | TensorBase< Derived > | |
base_sizes() const | TensorBase< Derived > | |
base_storage() const | TensorBase< Derived > | |
base_transpose(Size d1, Size d2) const | TensorBase< Derived > | |
base_unsqueeze(Size d) const | TensorBase< Derived > | |
batch_dim() const | TensorBase< Derived > | |
batch_expand(TensorShapeRef batch_size) const | TensorBase< Derived > | |
batch_expand_as(const Derived2 &other) const | TensorBase< Derived > | |
batch_expand_copy(TensorShapeRef batch_size) const | TensorBase< Derived > | |
batch_index(indexing::TensorIndicesRef indices) const | TensorBase< Derived > | |
batch_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other) | TensorBase< Derived > | |
batch_reshape(TensorShapeRef batch_shape) const | TensorBase< Derived > | |
batch_size(Size index) const | TensorBase< Derived > | |
batch_sizes() const | TensorBase< Derived > | |
batch_transpose(Size d1, Size d2) const | TensorBase< Derived > | |
batch_unsqueeze(Size d) const | TensorBase< Derived > | |
batched() const | TensorBase< Derived > | |
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) const | TensorBase< Derived > | |
const_base_dim | PrimitiveTensor< Derived, S > | static |
const_base_sizes | PrimitiveTensor< Derived, S > | inlinestatic |
const_base_storage | PrimitiveTensor< Derived, S > | inlinestatic |
detach() const | TensorBase< Derived > | |
doc() const | NEML2Object | inline |
drotate(const Rot &r) const | R2Base< R2 > | |
drotate(const R2 &R) const | R2Base< R2 > | |
empty(const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< Derived, S > | static |
empty(TensorShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< Derived, S > | static |
empty_like(const Derived &other) | TensorBase< Derived > | static |
expected_options() | FillR2 | static |
neml2::R2::fill(const Real &a, const torch::TensorOptions &options=default_tensor_options()) | R2Base< R2 > | static |
neml2::R2::fill(const Scalar &a) | R2Base< R2 > | static |
neml2::R2::fill(const Real &a11, const Real &a22, const Real &a33, const torch::TensorOptions &options=default_tensor_options()) | R2Base< R2 > | static |
neml2::R2::fill(const Scalar &a11, const Scalar &a22, const Scalar &a33) | R2Base< R2 > | static |
neml2::R2::fill(const Real &a11, const Real &a22, const Real &a33, const Real &a23, const Real &a13, const Real &a12, const torch::TensorOptions &options=default_tensor_options()) | R2Base< R2 > | static |
neml2::R2::fill(const Scalar &a11, const Scalar &a22, const Scalar &a33, const Scalar &a23, const Scalar &a13, const Scalar &a12) | R2Base< R2 > | static |
neml2::R2::fill(const Real &a11, const Real &a12, const Real &a13, const Real &a21, const Real &a22, const Real &a23, const Real &a31, const Real &a32, const Real &a33, const torch::TensorOptions &options=default_tensor_options()) | R2Base< R2 > | static |
neml2::R2::fill(const Scalar &a11, const Scalar &a12, const Scalar &a13, const Scalar &a21, const Scalar &a22, const Scalar &a23, const Scalar &a31, const Scalar &a32, const Scalar &a33) | R2Base< R2 > | static |
FillR2(const OptionSet &options) | FillR2 | |
full(Real init, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< Derived, S > | static |
full(TensorShapeRef batch_shape, Real init, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< Derived, S > | static |
full_like(const Derived &other, Real init) | TensorBase< Derived > | static |
host() const | NEML2Object | |
host() | NEML2Object | |
identity(const torch::TensorOptions &options=default_tensor_options()) | R2Base< R2 > | static |
identity_map(const torch::TensorOptions &options=default_tensor_options()) | R2 | static |
input_options() const | NEML2Object | inline |
inverse() const | R2Base< R2 > | |
linspace(const Derived &start, const Derived &end, Size nstep, Size dim=0, Size batch_dim=-1) | TensorBase< Derived > | static |
logspace(const Derived &start, const Derived &end, Size nstep, Size dim=0, Size batch_dim=-1, Real base=10) | TensorBase< Derived > | static |
name() const | NEML2Object | inline |
NEML2Object(const OptionSet &options) | NEML2Object | |
ones(const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< Derived, S > | static |
ones(TensorShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< Derived, S > | static |
ones_like(const Derived &other) | TensorBase< Derived > | static |
operator SR2() const | R2Base< R2 > | explicit |
operator Tensor() const | PrimitiveTensor< Derived, S > | |
operator()(Size i, Size j) const | R2Base< R2 > | |
operator-() const | TensorBase< Derived > | |
path() const | NEML2Object | inline |
PrimitiveTensor()=default | PrimitiveTensor< Derived, S > | |
PrimitiveTensor(const torch::Tensor &tensor, Size batch_dim) | PrimitiveTensor< Derived, S > | explicit |
PrimitiveTensor(const torch::Tensor &tensor) | PrimitiveTensor< Derived, S > | |
R2(const SR2 &S) | R2 | |
R2(const WR2 &W) | R2 | |
R2(const Rot &r) | R2 | explicit |
rotate(const Rot &r) const | R2Base< R2 > | |
rotate(const R2 &R) const | R2Base< R2 > | |
setup() | NEML2Object | inlinevirtual |
skew(const Vec &v) | R2Base< R2 > | static |
TensorBase()=default | TensorBase< Derived > | |
TensorBase(const torch::Tensor &tensor, Size batch_dim) | TensorBase< Derived > | |
TensorBase(const Derived &tensor) | TensorBase< Derived > | |
TensorBase(Real)=delete | TensorBase< Derived > | |
to(const torch::TensorOptions &options) const | TensorBase< Derived > | |
transpose() const | R2Base< R2 > | |
type() const | NEML2Object | inline |
UserTensorBase(const OptionSet &options) | UserTensorBase | |
zeros(const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< Derived, S > | static |
zeros(TensorShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< Derived, S > | static |
zeros_like(const Derived &other) | TensorBase< Derived > | static |
~NEML2Object()=default | NEML2Object | virtual |