NEML2 1.4.0
Loading...
Searching...
No Matches
LabeledMatrix Member List

This is the complete list of members for LabeledMatrix, including all inherited members.

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