def METHOD_NAME(self):
    """Get type-wise weight"""
    if self.regularizer is None:
        return self.W
    elif self.regularizer == "basis":
        W = self.W.view(self.num_bases, self.in_size * self.out_size)
        return (self.coeff @ W).view(
            self.num_types, self.in_size, self.out_size
        )
    elif self.regularizer == "bdd":
        return self.W
    else:
        raise ValueError(
            f'Supported regularizer options: "basis", "bdd", but got {regularizer}'
        )