NEML2 1.4.0
|
This is the complete list of members for FillSR2, including all inherited members.
base_dim() const | BatchTensorBase< Derived > | |
base_expand(TorchShapeRef base_size) const | BatchTensorBase< Derived > | |
base_expand_as(const Derived2 &other) const | BatchTensorBase< Derived > | |
base_expand_copy(TorchShapeRef base_size) const | BatchTensorBase< Derived > | |
base_index(const TorchSlice &indices) const | BatchTensorBase< Derived > | |
base_index_put(const TorchSlice &indices, const torch::Tensor &other) | BatchTensorBase< Derived > | |
base_movedim(TorchSize d1, TorchSize d2) const | BatchTensorBase< Derived > | |
base_reshape(TorchShapeRef base_shape) const | BatchTensorBase< Derived > | |
base_size(TorchSize index) const | BatchTensorBase< Derived > | |
base_sizes() const | BatchTensorBase< Derived > | |
base_storage() const | BatchTensorBase< Derived > | |
base_transpose(TorchSize d1, TorchSize d2) const | BatchTensorBase< Derived > | |
base_unsqueeze(TorchSize d) const | BatchTensorBase< Derived > | |
batch_dim() const | BatchTensorBase< Derived > | |
batch_dim() | BatchTensorBase< Derived > | |
batch_expand(TorchShapeRef batch_size) const | BatchTensorBase< Derived > | |
batch_expand_as(const Derived2 &other) const | BatchTensorBase< Derived > | |
batch_expand_copy(TorchShapeRef batch_size) const | BatchTensorBase< Derived > | |
batch_index(TorchSlice indices) const | BatchTensorBase< Derived > | |
batch_index_put(TorchSlice indices, const torch::Tensor &other) | BatchTensorBase< Derived > | |
batch_reshape(TorchShapeRef batch_shape) const | BatchTensorBase< Derived > | |
batch_size(TorchSize index) const | BatchTensorBase< Derived > | |
batch_sizes() const | BatchTensorBase< Derived > | |
batch_sum(TorchSize d) const | BatchTensorBase< Derived > | |
batch_transpose(TorchSize d1, TorchSize d2) const | BatchTensorBase< Derived > | |
batch_unsqueeze(TorchSize d) const | BatchTensorBase< Derived > | |
batched() const | BatchTensorBase< Derived > | |
BatchTensorBase()=default | BatchTensorBase< Derived > | |
BatchTensorBase(const torch::Tensor &tensor, TorchSize batch_dim) | BatchTensorBase< Derived > | |
BatchTensorBase(const Derived &tensor) | BatchTensorBase< Derived > | |
BatchTensorBase(Real)=delete | BatchTensorBase< Derived > | |
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) const | BatchTensorBase< Derived > | |
const_base_dim | FixedDimTensor< SR2, 6 > | static |
const_base_sizes | FixedDimTensor< SR2, 6 > | inlinestatic |
const_base_storage | FixedDimTensor< SR2, 6 > | inlinestatic |
det() const | SR2 | |
detach() const | BatchTensorBase< Derived > | |
dev() const | SR2 | |
doc() const | NEML2Object | inline |
drotate(const Rot &r) const | SR2 | |
drotate(const R2 &R) const | SR2 | |
empty(const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< SR2, 6 > | static |
empty(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< SR2, 6 > | static |
empty_like(const Derived &other) | BatchTensorBase< Derived > | static |
expected_options() | FillSR2 | static |
neml2::SR2::fill(const Real &a, const torch::TensorOptions &options=default_tensor_options()) | SR2 | static |
neml2::SR2::fill(const Scalar &a) | SR2 | static |
neml2::SR2::fill(const Real &a11, const Real &a22, const Real &a33, const torch::TensorOptions &options=default_tensor_options()) | SR2 | static |
neml2::SR2::fill(const Scalar &a11, const Scalar &a22, const Scalar &a33) | SR2 | static |
neml2::SR2::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()) | SR2 | static |
neml2::SR2::fill(const Scalar &a11, const Scalar &a22, const Scalar &a33, const Scalar &a23, const Scalar &a13, const Scalar &a12) | SR2 | static |
FillSR2(const OptionSet &options) | FillSR2 | |
FixedDimTensor()=default | FixedDimTensor< SR2, 6 > | |
FixedDimTensor(const torch::Tensor &tensor, TorchSize batch_dim) | FixedDimTensor< SR2, 6 > | explicit |
FixedDimTensor(const torch::Tensor &tensor) | FixedDimTensor< SR2, 6 > | |
full(Real init, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< SR2, 6 > | static |
full(TorchShapeRef batch_shape, Real init, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< SR2, 6 > | static |
full_like(const Derived &other, Real init) | BatchTensorBase< Derived > | static |
host() const | NEML2Object | |
host() | NEML2Object | |
identity(const torch::TensorOptions &options=default_tensor_options()) | SR2 | static |
identity_map(const torch::TensorOptions &options=default_tensor_options()) | SR2 | static |
inner(const SR2 &other) const | SR2 | |
input_options() const | NEML2Object | inline |
inverse() const | SR2 | |
linspace(const Derived &start, const Derived &end, TorchSize nstep, TorchSize dim=0, TorchSize batch_dim=-1) | BatchTensorBase< Derived > | static |
list_sum() const | BatchTensorBase< Derived > | |
list_unsqueeze() const | BatchTensorBase< Derived > | |
logspace(const Derived &start, const Derived &end, TorchSize nstep, TorchSize dim=0, TorchSize batch_dim=-1, Real base=10) | BatchTensorBase< Derived > | static |
name() const | NEML2Object | inline |
NEML2Object(const OptionSet &options) | NEML2Object | |
norm(Real eps=0) const | SR2 | |
norm_sq() const | SR2 | |
ones(const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< SR2, 6 > | static |
ones(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< SR2, 6 > | static |
ones_like(const Derived &other) | BatchTensorBase< Derived > | static |
operator BatchTensor() const | FixedDimTensor< SR2, 6 > | |
operator()(TorchSize i, TorchSize j) const | SR2 | |
operator-() const | BatchTensorBase< Derived > | |
outer(const SR2 &other) const | SR2 | |
path() const | NEML2Object | inline |
rotate(const Rot &r) const | SR2 | |
rotate(const R2 &R) const | SR2 | |
setup() | NEML2Object | inlinevirtual |
SR2(const R2 &T) | SR2 | |
to(const torch::TensorOptions &options) const | BatchTensorBase< Derived > | |
tr() const | SR2 | |
transpose() const | SR2 | |
type() const | NEML2Object | inline |
UserTensor(const OptionSet &options) | UserTensor | |
vol() const | SR2 | |
zeros(const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< SR2, 6 > | static |
zeros(TorchShapeRef batch_shape, const torch::TensorOptions &options=default_tensor_options()) | FixedDimTensor< SR2, 6 > | static |
zeros_like(const Derived &other) | BatchTensorBase< Derived > | static |
~NEML2Object()=default | NEML2Object | virtual |