diff --git a/torch_geometric/nn/dense/linear.py b/torch_geometric/nn/dense/linear.py index 66656fec9015..4fdaef0092bf 100644 --- a/torch_geometric/nn/dense/linear.py +++ b/torch_geometric/nn/dense/linear.py @@ -177,8 +177,8 @@ class HeteroLinear(torch.nn.Module): Shapes: - **input:** - features :math:`(*, F_{in})`, - type vector :math:`(*)` + features :math:`(*, F_{in})`, + type vector :math:`(*)` - **output:** features :math:`(*, F_{out})` """ def __init__(self, in_channels: int, out_channels: int, num_types: int,