Method embed

script/models/nerfw.py:128–133

(self, inputs)

        self.out_dim = out_dim

    def embed(self, inputs):
        if self.kwargs['max_freq_log2'] != 0:
            ret = torch.cat([fn(inputs) for fn in self.embed_fns], -1) # cos, sin embedding
        else:
            ret = inputs
        return ret

    def get_embed_weight(self, epoch, num_freqs, N):
        ''' Nerfie Paper Eq.(8) '''