Script.square¶
- Script.square(x, *, out=None)¶
Compute the element-wise square (x^2).
- Parameters:
x (RegisterTensor) – Input tensor.
out (RegisterTensor, optional) – Output tensor. If not provided, a new tensor is allocated.
- Returns:
ret – Tensor with the same shape and dtype as
x.- Return type:
Notes
Thread group: Can be executed by any sized thread group.