i am sorry,i don’t state my question clearly,the whole code is showed below:
import torch.nn as nn
import torch
class structural_attention_layer(nn.Module):
    def init(self):
        super(structural_attention_layer,self).init()

    def forward(self,a,b):
        return torch.randn(a,b)
    
s_att=nn.Sequential()
s_att.add_module(name="structural_attention_layer",module=structural_attention_layer())
s_att(1,2)