Function make_kernel

models/float/styledecoder.py:36–41

(k)

def make_kernel(k):
    k = torch.tensor(k, dtype=torch.float32)
    if k.ndim == 1:
        k = k[None, :] * k[:, None]
    k /= k.sum()
    return k


class FusedLeakyReLU(nn.Module):