Spec-Zone.ru › TensorFlow 2.9

tf.convert_to_tensor

Просмотреть исходный код на GitHub

Преобразует заданный value в Tensor.

tf.convert_to_tensor(
    value, dtype=None, dtype_hint=None, name=None
)

Эта функция преобразует объекты Python различных типов в объекты Tensor. Она принимает объекты Tensor, массивы NumPy, списки Python и скаляры Python.

Например:

import numpy as np
def my_func(arg):
  arg = tf.convert_to_tensor(arg, dtype=tf.float32)
  return arg
# The following calls are equivalent.

value_1 = my_func(tf.constant([[1.0, 2.0], [3.0, 4.0]]))
print(value_1)
tf.Tensor(
  [[1. 2.]
   [3. 4.]], shape=(2, 2), dtype=float32)
value_2 = my_func([[1.0, 2.0], [3.0, 4.0]])
print(value_2)
tf.Tensor(
  [[1. 2.]
   [3. 4.]], shape=(2, 2), dtype=float32)
value_3 = my_func(np.array([[1.0, 2.0], [3.0, 4.0]], dtype=np.float32))
print(value_3)
tf.Tensor(
  [[1. 2.]
   [3. 4.]], shape=(2, 2), dtype=float32)

Эта функция может быть полезной при составлении новой операции в Python (например, my_func в примере выше). Все стандартные конструкторы Python-операций применяют эту функцию к каждому из их входных значений типа Tensor, что позволяет этим операциям принимать массивы NumPy, списки Python и скаляры в дополнение к объектам Tensor.

Примечание: Данная функция отличается от стандартного поведения NumPy для типов float и string, когда None присутствует в списке Python или скаляре. Вместо безмолвной конвертации None значений будет генерироваться ошибка.
Аргументы
value Объект, чьий тип имеет зарегистрированную функцию конвертации в Tensor.
dtype Необязательный тип элемента для возвращаемого тензора. Если отсутствует, тип определяется по типу value.
dtype_hint Необязательный тип элемента для возвращаемого тензора, используется, когда dtype равен None. В некоторых случаях у вызывающего объекта может не быть типа dtype при преобразовании в тензор, поэтому dtype_hint может использоваться как мягкая предпочтительная настройка. Если преобразование в dtype_hint невозможно, этот аргумент не оказывает влияния.
name Необязательное имя для использования, если создается новый Tensor.
Возвращаемое значение
Tensor на основе value.
Возможные исключения
TypeError Если для value в dtype нет зарегистрированной функции конвертации.
RuntimeError Если зарегистрированная функция конвертации возвращает недопустимое значение.
ValueError Если value — это тензор, не соответствующий заданному типу dtype в режиме графа.

© 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/versions/r2.9/api_docs/python/tf/convert_to_tensor

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API