tf.math.square

Поэлементно вычисляет квадрат x.

Основные псевдонимы

tf.square

Псевдонимы для обеспечения совместимости при миграции

tf.math.square(
    x: Annotated[Any, tf.raw_ops.Any],
    name=None
) -> Annotated[Any, tf.raw_ops.Any]

Используется в руководстве	Используется в учебных материалах
Повышение производительности с помощью tf.function Основные циклы обучения Основы TensorFlow Распределенное обучение с Core API и DTensor Многослойные перцептроны для распознавания цифр с Core API	Основы настройки: тензоры и операции

То есть \(y = x * x = x^2\).

tf.math.square([-2., 0., 3.])
<tf.Tensor: shape=(3,), dtype=float32, numpy=array([4., 0., 9.], dtype=float32)>

Аргументы
`x`	Тензор `Tensor`. Должен иметь один из следующих типов: `bfloat16`, `half`, `float32`, `float64`, `int8`, `int16`, `int32`, `int64`, `uint8`, `uint16`, `uint32`, `uint64`, `complex64`, `complex128`.
`name`	Имя операции (необязательно).

Возвращает
Тензор `Tensor`. Имеет тот же тип, что и `x`. Если `x` — это `SparseTensor`, возвращает `SparseTensor(x.indices, tf.math.square(x.values, ...), x.dense_shape)`

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/math/square