NEML2 1.4.0
|
This is the complete list of members for LabeledMatrix, including all inherited members.
__init__(self) | LabeledMatrix | |
__init__(self, torch.Tensor arg0, typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] arg1) | LabeledMatrix | |
__init__(self, Tensor arg0, typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] arg1) | LabeledMatrix | |
__init__(self, LabeledMatrix arg0) | LabeledMatrix | |
__str__(self) | LabeledMatrix | |
axes(self) | LabeledMatrix | |
base(self) | LabeledMatrix | |
batch(self) | LabeledMatrix | |
batched(self) | LabeledMatrix | |
clone(self) | LabeledMatrix | |
copy_(self, torch.Tensor arg0) | LabeledMatrix | |
detach(self) | LabeledMatrix | |
detach_(self) | LabeledMatrix | |
device(self) | LabeledMatrix | |
dim(self) | LabeledMatrix | |
dtype(self) | LabeledMatrix | |
empty(tuple[int,...] batch_shape, typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] axes, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) | LabeledMatrix | static |
grad(self) | LabeledMatrix | |
requires_grad(self) | LabeledMatrix | |
requires_grad_(self, bool arg0) | LabeledMatrix | |
shape(self) | LabeledMatrix | |
tensor(self) | LabeledMatrix | |
to(self, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) | LabeledMatrix | |
torch(self) | LabeledMatrix | |
zero_(self) | LabeledMatrix | |
zeros(tuple[int,...] batch_shape, typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] axes, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) | LabeledMatrix | static |