Function angle_transform

models/encoders/DFormerv2.py:109–112 · view source on GitHub ↗

(x, sin, cos)

Source from the content-addressed store, hash-verified

107
108
109	def angle_transform(x, sin, cos):
110	x1 = x[:, :, :, :, ::2]
111	x2 = x[:, :, :, :, 1::2]
112	return (x * cos) + (torch.stack([-x2, x1], dim=-1).flatten(-2) * sin)
113
114
115	class GeoPriorGen(nn.Module):

forwardMethod · 0.85

no outgoing calls

no test coverage detected