src/features.py [395:404]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        inputs = ray_sample_positions
        inputs_flat = torch.reshape(inputs, [-1, inputs.shape[-1]])

        embedded = self.pos_enc.encode(inputs_flat)

        input_dirs = ray_directions[:, None].expand(inputs.shape)
        input_dirs_flat = torch.reshape(input_dirs, [-1, input_dirs.shape[-1]])
        embedded_dirs = self.dir_enc.encode(input_dirs_flat)

        embedded = torch.cat([embedded, embedded_dirs], -1)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/features.py [522:531]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        inputs = ray_sample_positions
        inputs_flat = torch.reshape(inputs, [-1, inputs.shape[-1]])

        embedded = self.pos_enc.encode(inputs_flat)

        input_dirs = ray_directions[:, None].expand(inputs.shape)
        input_dirs_flat = torch.reshape(input_dirs, [-1, input_dirs.shape[-1]])
        embedded_dirs = self.dir_enc.encode(input_dirs_flat)

        embedded = torch.cat([embedded, embedded_dirs], -1)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -