Hi,
I can use torch.pow like this:
factor = 2.3
a = torch.randn(2, 4, 2, 2)
b = a.pow(factor)

However, I need to use something like this:
factor = torch.tensor([2.1, 2.2, 2.3, 2.4])
a = torch.randn(2, 4, 2, 2)
b = a.pow(factor, dim=1)

It seems that pytorch does not support this. How could I do that along with its backward computation?