How to calculate the [batch_size, alphas] [batch, seq, vector]]

1

Hi:
how to calculate the A = [batch_size, alphas], B=[batch, seq, vector],
to get [batch_size, seq, alpha*vector]

2

A.unsqueeze(1).unsqueeze(3) + B.unsqueeze(2)