NEML2 1.4.0
|
This is the complete list of members for FullTensor, including all inherited members.
base_dim() const | TensorBase< Tensor > | |
base_expand(TensorShapeRef base_size) const | TensorBase< Tensor > | |
base_expand_as(const Derived2 &other) const | TensorBase< Tensor > | |
base_expand_copy(TensorShapeRef base_size) const | TensorBase< Tensor > | |
base_index(indexing::TensorIndicesRef indices) const | TensorBase< Tensor > | |
base_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other) | TensorBase< Tensor > | |
base_reshape(TensorShapeRef base_shape) const | TensorBase< Tensor > | |
base_size(Size index) const | TensorBase< Tensor > | |
base_sizes() const | TensorBase< Tensor > | |
base_storage() const | TensorBase< Tensor > | |
base_transpose(Size d1, Size d2) const | TensorBase< Tensor > | |
base_unsqueeze(Size d) const | TensorBase< Tensor > | |
batch_dim() const | TensorBase< Tensor > | |
batch_expand(TensorShapeRef batch_size) const | TensorBase< Tensor > | |
batch_expand_as(const Derived2 &other) const | TensorBase< Tensor > | |
batch_expand_copy(TensorShapeRef batch_size) const | TensorBase< Tensor > | |
batch_index(indexing::TensorIndicesRef indices) const | TensorBase< Tensor > | |
batch_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other) | TensorBase< Tensor > | |
batch_reshape(TensorShapeRef batch_shape) const | TensorBase< Tensor > | |
batch_size(Size index) const | TensorBase< Tensor > | |
batch_sizes() const | TensorBase< Tensor > | |
batch_transpose(Size d1, Size d2) const | TensorBase< Tensor > | |
batch_unsqueeze(Size d) const | TensorBase< Tensor > | |
batched() const | TensorBase< Tensor > | |
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) const | TensorBase< Tensor > | |
detach() const | TensorBase< Tensor > | |
doc() const | NEML2Object | inline |
empty(const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
empty(const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
empty_like(const Tensor &other) | TensorBase< Tensor > | static |
expected_options() | FullTensor | static |
full(const TensorShapeRef &base_shape, Real init, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
full(const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, Real init, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
full_like(const Tensor &other, Real init) | TensorBase< Tensor > | static |
FullTensor(const OptionSet &options) | FullTensor | |
host() const | NEML2Object | |
host() | NEML2Object | |
identity(Size n, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
identity(const TensorShapeRef &batch_shape, Size n, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
input_options() const | NEML2Object | inline |
linspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0, Size batch_dim=-1) | TensorBase< Tensor > | static |
logspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0, Size batch_dim=-1, Real base=10) | TensorBase< Tensor > | static |
name() const | NEML2Object | inline |
NEML2Object(const OptionSet &options) | NEML2Object | |
ones(const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
ones(const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
ones_like(const Tensor &other) | TensorBase< Tensor > | static |
operator-() const | TensorBase< Tensor > | |
path() const | NEML2Object | inline |
setup() | NEML2Object | inlinevirtual |
TensorBase()=default | TensorBase< Tensor > | |
TensorBase(const torch::Tensor &tensor, Size batch_dim) | TensorBase< Tensor > | |
TensorBase(const Tensor &tensor) | TensorBase< Tensor > | |
TensorBase(Real)=delete | TensorBase< Tensor > | |
to(const torch::TensorOptions &options) const | TensorBase< Tensor > | |
type() const | NEML2Object | inline |
UserTensorBase(const OptionSet &options) | UserTensorBase | |
zeros(const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
zeros(const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
zeros_like(const Tensor &other) | TensorBase< Tensor > | static |
~NEML2Object()=default | NEML2Object | virtual |