Repository URL to install this package:
|
Version:
2.0.0rc1 ▾
|
from ray.rllib.models.torch.modules.gru_gate import GRUGate
from ray.rllib.models.torch.modules.multi_head_attention import MultiHeadAttention
from ray.rllib.models.torch.modules.relative_multi_head_attention import (
RelativeMultiHeadAttention,
)
from ray.rllib.models.torch.modules.skip_connection import SkipConnection
__all__ = [
"GRUGate",
"RelativeMultiHeadAttention",
"SkipConnection",
"MultiHeadAttention",
]