| """公用算子模块集合。""" |
|
|
| from __future__ import annotations |
|
|
| from .ffn import SwiGLUFFN |
| from .gate_attention import GateSelfAttention, GateCrossAttention |
| from .moe import MoEBlock, PerLayerExperts |
| from .normalization import symlog, symexp |
| from .pos_encoding import RoPE3D, build_rope_freqs |
| from .learned_pe import LearnedTokenPE |
| from .rays import FThetaCamera, compute_ego_rays |
| from .temporal_compress import TemporalCompress2x2x2 |
|
|
| __all__ = [ |
| "SwiGLUFFN", |
| "GateSelfAttention", |
| "GateCrossAttention", |
| "MoEBlock", |
| "PerLayerExperts", |
| "symlog", |
| "symexp", |
| "RoPE3D", |
| "build_rope_freqs", |
| "LearnedTokenPE", |
| "FThetaCamera", |
| "compute_ego_rays", |
| "TemporalCompress2x2x2", |
| ] |
|
|
