pytorch 不同数据循环经过同一个linear层,可以正常更新参数吗

如题,在pytorch框架中,假如现在有shape为(2,3,4)的tensor,2为batch size,3为目标数量,每个目标是4维向量。

现在想要让这三个目标向量先后通过同一个nn.Linear层,然后cat在一起后进入接下来的其他层,这样可以吗?

在梯度回传的时候,该线性层的参数会更新三次吗?

https://blog.csdn.net/mangobar/article/details/102606792