class_docstr: Network that predicts warps as an SE(3) field.
  
  Attributes:
    points_encoder: the positional encoder for the points.
    metadata_encoder: an encoder for metadata.
    alpha: the alpha for the positional encoding.
    skips: the index of the layers with skip connections.
    depth: the depth of the network excluding the logit layer.
    hidden_channels: the width of the network hidden layers.
    activation: the activation for each layer.
    metadata_encoded: whether the metadata parameter is pre-encoded or not.
    hidden_initializer: the initializer for the hidden layers.
    output_initializer: the initializer for the last logit layer.
class_docstr_tok: ["Network","that","predicts","warps","as","an","SE","(","3",")","field",".","Attributes",":","points_encoder",":","the","positional","encoder","for","the","points",".","metadata_encoder",":","an","encoder","for","metadata",".","alpha",":","the","alpha","for","the","positional","encoding",".","skips",":","the","index","of","the","layers","with","skip","connections",".","depth",":","the","depth","of","the","network","excluding","the","logit","layer",".","hidden_channels",":","the","width","of","the","network","hidden","layers",".","activation",":","the","activation","for","each","layer",".","metadata_encoded",":","whether","the","metadata","parameter","is","pre-encoded","or","not",".","hidden_initializer",":","the","initializer","for","the","hidden","layers",".","output_initializer",":","the","initializer","for","the","last","logit","layer","."]
code: def encode_metadata(
      self, metadata: jnp.ndarray, time_alpha: Optional[float] = None
  ):
      if self.metadata_encoder_type == "time":
          metadata_embed = self.metadata_encoder(metadata, time_alpha)
      elif self.metadata_encoder_type == "glo":
          metadata_embed = self.metadata_encoder(metadata)
      else:
          raise RuntimeError(
              f"Unknown metadata encoder type {self.metadata_encoder_type}"
          )
  
      return metadata_embed
  
imports: from typing import Any, Iterable, Optional, Dict
  from flax import linen
  import jax
  import jax.numpy
  from nerfies import glo
  from nerfies import model_utils
  from nerfies import modules
  from nerfies import rigid_body
  from nerfies import types
label_desc: Use image node_id 2 for calling the SE3Field obj's underlying member method code with example usage: obj.encode_metadata(metadata, time_alpha) and returns: metadata_embed