Функция make_tensor_proto

Функция make_tensor_proto преобразует список, число или массив в объект TensorProto - низкоуровневое представление тензора, которое используется в графах TensorFlow и при сериализации. Первым параметром передается значение, вторым - тип данных, третьим - форма тензора.

Синтаксис

tf.make_tensor_proto(values, dtype=None, shape=None)

Пример

Давайте создадим TensorProto из списка чисел 1, 2, 3, 4, 5:

import tensorflow as tf

proto = tf.make_tensor_proto([1, 2, 3, 4, 5])
print(proto)

Результат выполнения кода:

dtype: DT_INT32
tensor_shape {
  dim {
	size: 5
  }
}
tensor_content: "\001\000\000\000\002\000\000\000\003\000\000\000\004\000\000\000\005\000\000\000"

Пример

Давайте создадим TensorProto с явным указанием типа tf.float32:

import tensorflow as tf

proto = tf.make_tensor_proto([1, 2, 3, 4, 5], dtype=tf.float32)
print(proto.dtype)

Результат выполнения кода:

DT_FLOAT

Пример

Давайте создадим двумерный TensorProto и проверим его форму:

import tensorflow as tf

proto = tf.make_tensor_proto([[1, 2, 3], [4, 5, 6]])
print(proto.tensor_shape)

Результат выполнения кода:

dim {
  size: 2
}
dim {
  size: 3
}

Смотрите также

функцию convert_to_tensor,
которая преобразует данные в тензор TensorFlow
функцию constant,
которая создает тензор из переданных данных
функцию make_ndarray,
которая преобразует TensorProto в массив NumPy
класс TensorSpec,
который описывает тип и форму тензора