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(TorchSize i=0) constLabeledTensor< Derived, D >inline
base_dim() constLabeledTensor< Derived, D >
base_index(TorchSlice indices) constLabeledTensor< Derived, D >
base_index_put(TorchSlice indices, const torch::Tensor &other)LabeledTensor< Derived, D >
base_sizes() constLabeledTensor< Derived, D >
batch_dim() constLabeledTensor< Derived, D >
batch_index(TorchSlice indices) constLabeledTensor< Derived, D >
batch_index_put(TorchSlice indices, const torch::Tensor &other)LabeledTensor< Derived, D >
batch_sizes() constLabeledTensor< Derived, D >
block(S &&... names) constLabeledTensor< Derived, D >
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) constLabeledTensor< Derived, D >
copy_(const T &other)LabeledTensor< Derived, D >
detach() constLabeledTensor< Derived, D >
detach_()LabeledTensor< Derived, D >
empty(TorchShapeRef batch_shape, const std::vector< const LabeledAxis * > &axes, const torch::TensorOptions &options=default_tensor_options())LabeledTensor< Derived, D >static
empty_like(const Derived &other)LabeledTensor< Derived, D >static
get(S &&... names) constLabeledTensor< Derived, D >inline
get_list(S &&... names) constLabeledTensor< Derived, D >inline
LabeledTensor()=defaultLabeledTensor< Derived, D >
LabeledTensor(const torch::Tensor &tensor, TorchSize batch_dim, const std::vector< const LabeledAxis * > &axes)LabeledTensor< Derived, D >
LabeledTensor(const BatchTensor &tensor, const std::vector< const LabeledAxis * > &axes)LabeledTensor< Derived, D >
LabeledTensor(const Derived &other)LabeledTensor< Derived, D >
operator BatchTensor() constLabeledTensor< Derived, D >
operator torch::Tensor() constLabeledTensor< Derived, D >
operator()(S &&... names) constLabeledTensor< Derived, D >
operator-() constLabeledTensor< Derived, D >
operator=(const Derived &other)LabeledTensor< Derived, D >
options() constLabeledTensor< Derived, D >inline
set(const BatchTensorBase< T > &value, S &&... names)LabeledTensor< Derived, D >inline
set_list(const BatchTensorBase< T > &value, S &&... names)LabeledTensor< Derived, D >inline
slice(TorchSize i, const std::string &name) constLabeledTensor< Derived, D >
slice_indices(S &&... names) constLabeledTensor< Derived, D >
storage_size() constLabeledTensor< Derived, D >
storage_size(S &&... names) 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(TorchShapeRef batch_shape, const std::vector< const LabeledAxis * > &axes, const torch::TensorOptions &options=default_tensor_options())LabeledTensor< Derived, D >static
zeros_like(const Derived &other)LabeledTensor< Derived, D >static