NEML2 2.0.0
|
This is the complete list of members for WSR4, including all inherited members.
base_dim() const | TensorBase< Derived > | |
base_expand(TensorShapeRef base_shape) const | TensorBase< Derived > | |
base_expand(Size base_size, Size dim) const | TensorBase< Derived > | |
base_expand_as(const Derived2 &other) const | TensorBase< Derived > | |
base_expand_copy(TensorShapeRef base_shape) const | TensorBase< Derived > | |
base_flatten() const | TensorBase< Derived > | |
base_index(indexing::TensorIndicesRef indices) const | TensorBase< Derived > | |
base_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other) | TensorBase< Derived > | |
base_index_put_(indexing::TensorIndicesRef indices, Real v) | 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(const TraceableTensorShape &batch_shape) const | TensorBase< Derived > | |
batch_expand(const TraceableSize &batch_size, Size dim) const | TensorBase< Derived > | |
batch_expand_as(const Derived2 &other) const | TensorBase< Derived > | |
batch_expand_copy(const TraceableTensorShape &batch_shape) const | TensorBase< Derived > | |
batch_index(indexing::TensorIndicesRef indices) const | TensorBase< Derived > | |
batch_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other) | TensorBase< Derived > | |
batch_index_put_(indexing::TensorIndicesRef indices, Real v) | TensorBase< Derived > | |
batch_reshape(const TraceableTensorShape &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< WSR4, 3, 6 > | static |
const_base_sizes | PrimitiveTensor< WSR4, 3, 6 > | inlinestatic |
const_base_storage | PrimitiveTensor< WSR4, 3, 6 > | inlinestatic |
detach() const | TensorBase< Derived > | |
empty(const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WSR4, 3, 6 > | static |
empty(const TraceableTensorShape &batch_shape, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WSR4, 3, 6 > | static |
empty_like(const Derived &other) | TensorBase< Derived > | static |
full(Real init, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WSR4, 3, 6 > | static |
full(const TraceableTensorShape &batch_shape, Real init, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WSR4, 3, 6 > | static |
full_like(const Derived &other, Real init) | TensorBase< Derived > | static |
identity_map(const torch::TensorOptions &) | PrimitiveTensor< WSR4, 3, 6 > | inlinestatic |
linspace(const Derived &start, const Derived &end, Size nstep, Size dim=0) | TensorBase< Derived > | static |
logspace(const Derived &start, const Derived &end, Size nstep, Size dim=0, Real base=10) | TensorBase< Derived > | static |
ones(const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WSR4, 3, 6 > | static |
ones(const TraceableTensorShape &batch_shape, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WSR4, 3, 6 > | static |
ones_like(const Derived &other) | TensorBase< Derived > | static |
operator Tensor() const | PrimitiveTensor< WSR4, 3, 6 > | |
operator-() const | TensorBase< Derived > | |
PrimitiveTensor()=default | PrimitiveTensor< WSR4, 3, 6 > | |
PrimitiveTensor(const torch::Tensor &tensor, Size batch_dim) | PrimitiveTensor< WSR4, 3, 6 > | explicit |
PrimitiveTensor(const torch::Tensor &tensor, const TraceableTensorShape &batch_shape) | PrimitiveTensor< WSR4, 3, 6 > | explicit |
PrimitiveTensor(const TensorBase< Derived2 > &tensor) | PrimitiveTensor< WSR4, 3, 6 > | |
PrimitiveTensor(const torch::Tensor &tensor) | PrimitiveTensor< WSR4, 3, 6 > | |
TensorBase()=default | TensorBase< Derived > | |
TensorBase(const torch::Tensor &tensor, Size batch_dim) | TensorBase< Derived > | |
TensorBase(const torch::Tensor &tensor, const TraceableTensorShape &batch_shape) | TensorBase< Derived > | |
TensorBase(const TensorBase< Derived2 > &tensor) | TensorBase< Derived > | |
TensorBase(Real)=delete | TensorBase< Derived > | |
to(const torch::TensorOptions &options) const | TensorBase< Derived > | |
variable_data() const | TensorBase< Derived > | |
WSR4(const R4 &F) | WSR4 | |
zeros(const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WSR4, 3, 6 > | static |
zeros(const TraceableTensorShape &batch_shape, const torch::TensorOptions &options=default_tensor_options()) | PrimitiveTensor< WSR4, 3, 6 > | static |
zeros_like(const Derived &other) | TensorBase< Derived > | static |