本文介绍了PyTorch中的square()函数，该函数可对0D及多维张量的元素进行平方运算。示例展示了如何处理整数、浮点数、复数和布尔值等不同类型的张量。
完成下面两步后，将自动完成登录并继续当前操作。
