Update the input tensor (torsion angles). Args: x (Tensor): input tensor x_score (Tensor): input tensor's score t (Tensor): timesteps tensor dt (Tensor): timestep size tensor x_mask (Tensor): input mask tensor Returns:
(self, x, x_score, t, dt, x_mask=None)
| 117 | |
| 118 | @abstractmethod |
| 119 | def step(self, x, x_score, t, dt, x_mask=None): |
| 120 | """Update the input tensor (torsion angles). |
| 121 | |
| 122 | Args: |
| 123 | x (Tensor): input tensor |
| 124 | x_score (Tensor): input tensor's score |
| 125 | t (Tensor): timesteps tensor |
| 126 | dt (Tensor): timestep size tensor |
| 127 | x_mask (Tensor): input mask tensor |
| 128 | |
| 129 | Returns: |
| 130 | Tensor: updated input tensor |
| 131 | """ |
| 132 | pass |
| 133 | |
| 134 | @abstractmethod |
| 135 | def sample_train_t(self, shape): |