NEML2 2.0.0
|
Public Member Functions | |
Vec | __add__ (self, float arg0) |
Vec | __add__ (self, Scalar arg0) |
Vec | __add__ (self, Vec arg0) |
Scalar | __call__ (self, int arg0) |
Vec | __iadd__ (self, float arg0) |
Vec | __imul__ (self, float arg0) |
None | __init__ (self) |
None | __init__ (self, torch.Tensor arg0, int arg1) |
None | __init__ (self, Vec arg0) |
None | __init__ (self, torch.Tensor arg0) |
None | __init__ (self, Rot arg0) |
Vec | __isub__ (self, float arg0) |
Vec | __itruediv__ (self, float arg0) |
Vec | __mul__ (self, float arg0) |
Vec | __mul__ (self, Scalar arg0) |
Vec | __neg__ (self) |
Vec | __pow__ (self, float arg0) |
Vec | __pow__ (self, Scalar arg0) |
Vec | __radd__ (self, float arg0) |
str | __repr__ (self) |
Vec | __rmul__ (self, float arg0) |
Tensor | __rpow__ (self, float arg0) |
Vec | __rsub__ (self, float arg0) |
Vec | __rtruediv__ (self, float arg0) |
str | __str__ (self) |
Vec | __sub__ (self, float arg0) |
Vec | __sub__ (self, Scalar arg0) |
Vec | __sub__ (self, Vec arg0) |
Vec | __truediv__ (self, float arg0) |
Vec | __truediv__ (self, Scalar arg0) |
torch.Tensor | __truediv__ (self, Vec arg0) |
bool | batched (self) |
Vec | clone (self) |
torch.Tensor | copy_ (self, torch.Tensor arg0, bool arg1) |
Vec | cross (self, Vec arg0) |
bool | defined (self) |
Vec | detach (self) |
torch.Tensor | detach_ (self) |
int | dim (self) |
Scalar | dot (self, Vec arg0) |
R2 | drotate (self, Rot arg0) |
Number|torch.SymInt|torch.SymFloat|torch.SymBool | item (self) |
Scalar | norm (self) |
Scalar | norm_sq (self) |
R2 | outer (self, Vec arg0) |
torch.Tensor | requires_grad_ (self, bool arg0) |
Vec | rotate (self, Rot arg0) |
Tensor | tensor (self) |
Vec | to (self, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
torch.Tensor | torch (self) |
torch.Tensor | torch (self) |
Vec | transform (self, R2 arg0) |
torch.Tensor | zero_ (self) |
VecBaseView | base (self) |
VecBatchView | 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 | |
Vec | empty (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | empty (tuple[int,...] batch_shape, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | empty_like (Vec arg0) |
Vec | fill (float x, float y, float z, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | fill (Scalar x, Scalar y, Scalar z) |
Vec | full (float fill_value, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | full (tuple[int,...] batch_shape, float fill_value, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | full_like (Vec arg0, float arg1) |
R2 | identity_map (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | linspace (Vec start, Vec end, int nstep, int dim=0) |
Vec | logspace (Vec start, Vec end, int nstep, int dim=0, float base=10.0) |
Vec | ones (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | ones (tuple[int,...] batch_shape, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | ones_like (Vec arg0) |
Vec | zeros (*, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | zeros (tuple[int,...] batch_shape, *, torch.dtype dtype=..., torch.device device=..., bool requires_grad=False) |
Vec | zeros_like (Vec arg0) |
None __init__ | ( | self | ) |
None __init__ | ( | self, | |
torch.Tensor | arg0, | ||
int | arg1 ) |
None __init__ | ( | self, | |
Vec | arg0 ) |
None __init__ | ( | self, | |
torch.Tensor | arg0 ) |
None __init__ | ( | self, | |
Rot | arg0 ) |
Vec __add__ | ( | self, | |
float | arg0 ) |
Scalar __call__ | ( | self, | |
int | arg0 ) |
Vec __iadd__ | ( | self, | |
float | arg0 ) |
Vec __imul__ | ( | self, | |
float | arg0 ) |
Vec __isub__ | ( | self, | |
float | arg0 ) |
Vec __itruediv__ | ( | self, | |
float | arg0 ) |
Vec __mul__ | ( | self, | |
float | arg0 ) |
Vec __neg__ | ( | self | ) |
Vec __pow__ | ( | self, | |
float | arg0 ) |
Vec __radd__ | ( | self, | |
float | arg0 ) |
str __repr__ | ( | self | ) |
Vec __rmul__ | ( | self, | |
float | arg0 ) |
Tensor __rpow__ | ( | self, | |
float | arg0 ) |
Vec __rsub__ | ( | self, | |
float | arg0 ) |
Vec __rtruediv__ | ( | self, | |
float | arg0 ) |
str __str__ | ( | self | ) |
Vec __sub__ | ( | self, | |
float | arg0 ) |
Vec __truediv__ | ( | self, | |
float | arg0 ) |
torch.Tensor __truediv__ | ( | self, | |
Vec | arg0 ) |
VecBaseView base | ( | self | ) |
VecBatchView batch | ( | self | ) |
bool batched | ( | self | ) |
Vec clone | ( | self | ) |
torch.Tensor copy_ | ( | self, | |
torch.Tensor | arg0, | ||
bool | arg1 ) |
bool defined | ( | self | ) |
Vec detach | ( | self | ) |
torch.Tensor detach_ | ( | self | ) |
torch.device device | ( | self | ) |
int dim | ( | self | ) |
torch.dtype dtype | ( | self | ) |
|
static |
|
static |
|
static |
|
static |
|
static |
torch.Tensor grad | ( | self | ) |
|
static |
Number | torch.SymInt | torch.SymFloat | torch.SymBool item | ( | self | ) |
Scalar norm | ( | self | ) |
Scalar norm_sq | ( | self | ) |
|
static |
|
static |
bool requires_grad | ( | self | ) |
torch.Tensor requires_grad_ | ( | self, | |
bool | arg0 ) |
tuple[int, ...] shape | ( | self | ) |
Tensor tensor | ( | self | ) |
Vec to | ( | self, | |
* | , | ||
torch.dtype | dtype = ..., | ||
torch.device | device = ..., | ||
bool | requires_grad = False ) |
torch.Tensor torch | ( | self | ) |
torch.Tensor torch | ( | self | ) |
torch.Tensor zero_ | ( | self | ) |
|
static |
|
static |