Spec-Zone.ru › TensorFlow 2.4

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-операций применяют эту функцию к каждому из их тензорных входных данных, что позволяет этим операциям принимать массивы 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 в режиме графа.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/convert_to_tensor

Spec-Zone.ru

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