NEML2 1.4.0
Loading...
Searching...
No Matches
LabeledTensor< Derived, D > Member List

This is the complete list of members for LabeledTensor< Derived, D >, including all inherited members.

_axesLabeledTensor< Derived, D >protected
_tensorLabeledTensor< Derived, D >protected
axes() constLabeledTensor< Derived, D >inline
axis(Size i=0) constLabeledTensor< Derived, D >inline
base_dim() constLabeledTensor< Derived, D >inline
base_index(indexing::TensorLabelsRef labels) constLabeledTensor< Derived, D >
base_index_put_(indexing::TensorLabelsRef labels, const Tensor &other)LabeledTensor< Derived, D >
base_size(Size d) constLabeledTensor< Derived, D >
base_sizes() constLabeledTensor< Derived, D >
base_storage() constLabeledTensor< Derived, D >
batch_dim() constLabeledTensor< Derived, D >
batch_index(indexing::TensorIndicesRef indices) constLabeledTensor< Derived, D >
batch_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other)LabeledTensor< Derived, D >
batch_size(Size d) constLabeledTensor< Derived, D >
batch_sizes() constLabeledTensor< Derived, D >
batched() constLabeledTensor< Derived, D >
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) constLabeledTensor< Derived, D >
copy_(const torch::Tensor &other)LabeledTensor< Derived, D >
detach() constLabeledTensor< Derived, D >
detach_()LabeledTensor< Derived, D >
device() constLabeledTensor< Derived, D >
dim() constLabeledTensor< Derived, D >
empty(TensorShapeRef batch_shape, const std::array< const LabeledAxis *, D > &axes, const torch::TensorOptions &options=default_tensor_options())LabeledTensor< Derived, D >static
LabeledTensor()=defaultLabeledTensor< Derived, D >
LabeledTensor(const torch::Tensor &tensor, const std::array< const LabeledAxis *, D > &axes)LabeledTensor< Derived, D >
LabeledTensor(const Tensor &tensor, const std::array< const LabeledAxis *, D > &axes)LabeledTensor< Derived, D >
LabeledTensor(const Derived &other)LabeledTensor< Derived, D >
operator Tensor() constLabeledTensor< Derived, D >
operator torch::Tensor() constLabeledTensor< Derived, D >
operator-() constLabeledTensor< Derived, D >
operator=(const Derived &other)LabeledTensor< Derived, D >
options() constLabeledTensor< Derived, D >
reinterpret(indexing::TensorLabelsRef indices) constLabeledTensor< Derived, D >
requires_grad() constLabeledTensor< Derived, D >
requires_grad_(bool req=true)LabeledTensor< Derived, D >
scalar_type() constLabeledTensor< Derived, D >
size(Size dim) constLabeledTensor< Derived, D >
sizes() constLabeledTensor< Derived, D >
tensor() constLabeledTensor< Derived, D >inline
tensor()LabeledTensor< Derived, D >inline
to(const torch::TensorOptions &options) constLabeledTensor< Derived, D >
zero_()LabeledTensor< Derived, D >
zeros(TensorShapeRef batch_shape, const std::array< const LabeledAxis *, D > &axes, const torch::TensorOptions &options=default_tensor_options())LabeledTensor< Derived, D >static