Square

In-place square.

Abstract Signature:

Square_(input: Tensor)

PyTorchJAX (Core)

PyTorch

API: torch.square_

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro 'jax.numpy.square({input})'