|
NEML2 2.0.0
|
The interface for working with tensor values (parameters, buffers, etc.) managed by models.
Public Member Functions | |
| None | requires_grad_ (self, bool req=True) |
| None | set_ (self, Tensor arg0) |
| Tensor | tensor (self) |
| torch.Tensor | torch (self) |
| torch.Tensor | grad (self) |
| bool | requires_grad (self) |
| torch.Tensor grad | ( | self | ) |
Retrieve the accumulated vector-Jacobian product after a backward propagation.
| bool requires_grad | ( | self | ) |
Value of the boolean requires_grad property of the underlying tensor.
| None requires_grad_ | ( | self, | |
| bool | req = True ) |
Set the requires_grad property of the underlying tensor.
| None set_ | ( | self, | |
| Tensor | arg0 ) |
Modify the underlying tensor data.
| Tensor tensor | ( | self | ) |
Convert to a tensors.Tensor
| torch.Tensor torch | ( | self | ) |
Convert to a torch.Tensor