Element-wise multiplication of a vector and a matrix

1

I’m searching the net for a multiplication which is applied between an 1-d tensor anf an n-d tensor.
Expected behaviour:

A = torch.tensor(np.array([[[1., 0., 0.],
                            [0., 1., 0.],
                            [0., 0., 1.]],
                           [[2., 0., 0.],
                            [0., 2., 0.],
                            [0., 0., 2.]]]))

v = torch.tensor(np.array([2.0, 1.0]))

torch.element_wise(v, A)

expected result:

tensor([[[2., 0., 0.],
         [0., 2., 0.],
         [0., 0., 2.]],
        [[2., 0., 0.],
         [0., 2., 0.],
         [0., 0., 2.]]])