NEML2 1.4.0
Loading...
Searching...
No Matches
LabeledMatrix Class Reference

Public Member Functions

None __init__ (self)
 
None __init__ (self, torch.Tensor arg0, typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] arg1)
 
None __init__ (self, Tensor arg0, typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] arg1)
 
None __init__ (self, LabeledMatrix arg0)
 
str __str__ (self)
 
bool batched (self)
 
LabeledMatrix clone (self)
 
None copy_ (self, torch.Tensor arg0)
 
LabeledMatrix detach (self)
 
None detach_ (self)
 
int dim (self)
 
None requires_grad_ (self, bool arg0)
 
Tensor tensor (self)
 
LabeledMatrix to (self, *torch.dtype dtype=..., torch.device device=..., bool requires_grad=False)
 
torch.Tensor torch (self)
 
None zero_ (self)
 
typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] axes (self)
 
LabeledMatrixBaseView base (self)
 
LabeledMatrixBatchView batch (self)
 
torch.device device (self)
 
torch.dtype dtype (self)
 
torch.Tensor grad (self)
 
bool requires_grad (self)
 
tuple[int,...] shape (self)
 

Static Public Member Functions

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 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)
 

Constructor & Destructor Documentation

◆ __init__() [1/4]

None __init__ ( self)

◆ __init__() [2/4]

None __init__ ( self,
torch.Tensor arg0,
typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] arg1 )

◆ __init__() [3/4]

None __init__ ( self,
Tensor arg0,
typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] arg1 )

◆ __init__() [4/4]

None __init__ ( self,
LabeledMatrix arg0 )

Member Function Documentation

◆ __str__()

str __str__ ( self)

◆ axes()

typing_extensions.Annotated[list[LabeledAxis], pybind11_stubgen.typing_ext.FixedSize(2)] axes ( self)

◆ base()

LabeledMatrixBaseView base ( self)

◆ batch()

LabeledMatrixBatchView batch ( self)

◆ batched()

bool batched ( self)

◆ clone()

LabeledMatrix clone ( self)

◆ copy_()

None copy_ ( self,
torch.Tensor arg0 )

◆ detach()

LabeledMatrix detach ( self)

◆ detach_()

None detach_ ( self)

◆ device()

torch.device device ( self)

◆ dim()

int dim ( self)

◆ dtype()

torch.dtype dtype ( self)

◆ empty()

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 )
static

◆ grad()

torch.Tensor grad ( self)

◆ requires_grad()

bool requires_grad ( self)

◆ requires_grad_()

None requires_grad_ ( self,
bool arg0 )

◆ shape()

tuple[int, ...] shape ( self)

◆ tensor()

Tensor tensor ( self)

◆ to()

LabeledMatrix to ( self,
*torch.dtype dtype = ...,
torch.device device = ...,
bool requires_grad = False )

◆ torch()

torch.Tensor torch ( self)

◆ zero_()

None zero_ ( self)

◆ zeros()

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 )
static