Like this, first I create a class in a.py, and I import b.py.
I want to pass the embedding to b.py through function    call(self.embedding)
class A(nn.Module):
	def __init__(self):
		self.embedding = nn.Embedding(3, 4)

	def forward(self):
		call(self.embedding)

In b.py, like this, I hope in class B, the parameter embedding can also be update.
class B(nn.Module):
	def __init__(self):

	def forward(self, embedding):

modelB = B()
def call(embedding):
	modelB(embedding)