Method _square_plus

ioblocks.py:37–38

(self, x)

        self.fc_weight = nn.Linear(c.dim, c.num_components)
    
    def _square_plus(self, x):
        return (x + T.sqrt(T.square(x) + 4)) / 2
    
    def input(self, sampled_latents: T.Tensor) -> T.Tensor:
        """Pre-sampled latents T.Tensor (B, L, Z) -> float tensor (B, L, D)"""