ViewAsRealΒΆ
Views a complex tensor as a real tensor.
Abstract Signature:
ViewAsReal(x: Tensor)
JAX (Core)
API:
βStrategy: Macro 'jax.numpy.stack([jax.numpy.real({x}), jax.numpy.imag({x})], axis=-1)'
NumPy
API:
βStrategy: Macro 'numpy.stack([numpy.real({x}), numpy.imag({x})], axis=-1)'
Keras
API:
βStrategy: Custom / Partial
TensorFlow
API:
tf.math.realStrategy: Macro 'tf.stack([tf.math.real({x}), tf.math.imag({x})], axis=-1)'
Apple MLX
API:
βStrategy: Custom / Partial
Flax NNX
API:
βStrategy: Macro 'jax.numpy.stack([jax.numpy.real({x}), jax.numpy.imag({x})], axis=-1)'
PaxML / Praxis
API:
βStrategy: Macro 'jax.numpy.stack([jax.numpy.real({x}), jax.numpy.imag({x})], axis=-1)'