Add noise to the input tensor (torsion angles). Args: x (Tensor): input tensor t (Tensor): timesteps tensor x_mask (Tensor): input mask tensor Returns: Tensor: noisy input tensor
(self, x, t, x_mask=None)
| 103 | |
| 104 | @abstractmethod |
| 105 | def add_noise(self, x, t, x_mask=None): |
| 106 | """Add noise to the input tensor (torsion angles). |
| 107 | |
| 108 | Args: |
| 109 | x (Tensor): input tensor |
| 110 | t (Tensor): timesteps tensor |
| 111 | x_mask (Tensor): input mask tensor |
| 112 | |
| 113 | Returns: |
| 114 | Tensor: noisy input tensor |
| 115 | """ |
| 116 | pass |
| 117 | |
| 118 | @abstractmethod |
| 119 | def step(self, x, x_score, t, dt, x_mask=None): |
nothing calls this directly
no outgoing calls
no test coverage detected