shlogg · Early preview
Super Kai (Kazuya Ito) @superkai_kazuya

Understanding PyTorch's Square() Function For Tensors

torch.square()` returns squared values for tensors with various dimensions & data types (ints, floats, complex nums, bools). Example: `torch.tensor([-3])` becomes `tensor(9)`.

Buy Me a Coffee☕
*Memos:

My post explains pow().
My post explains float_power().
My post explains abs() and sqrt().
My post explains gcd() and lcm().
My post explains trace(), reciprocal() and rsqrt().

square() can get the 0D or more D tensor of squared zero or more elements, getting the 0D or more D tensor of zero or more elements as shown below:
*Memos:

square() can be used with torch or a tensor.
The 1st argument(input) with torch or using a tensor(Required-Type:tensor of int, float, complex or bool).
There is out argument with torch(Optional-Default:None-Type:tensor):
*Memos:

out= mus...