How to calculate the square of a tensor?

1

just like numpy.square in numpy.
Thank you very much

2

Assume a tensor A, the the square of A equals A * A, that is B = A * A

3

You should also be able to do A**2 or A.pow(2)

1 Like
4

There is also a torch.square function now (PR) that you can call like A.square().