Skip to content

Instantly share code, notes, and snippets.

@radi-cho

radi-cho/gmlp_v_proj.py

Created September 10, 2022 20:06
Show Gist options
  • Save radi-cho/6b14a7ffbd1f42c127cb9043fad4b93a to your computer and use it in GitHub Desktop.
Save radi-cho/6b14a7ffbd1f42c127cb9043fad4b93a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gmlp_v_proj.py
v_channels = tf.linalg.matrix_transpose(v)
v_projected = self.spatial_projection(v_channels)
v_projected = tf.linalg.matrix_transpose(v_projected)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment