Как определить переменные и константы в TensorFlow?

Пользователь

от magdalen_kub , в категории: Другие , год назад

Как определить переменные и константы в TensorFlow?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от terrence , год назад

@magdalen_kub 

В TensorFlow переменные и константы определяются с помощью соответствующих функций из модуля tf:


Для определения константы используйте функцию tf.constant():

1
2
3
4
import tensorflow as tf

# Определение константы
my_constant = tf.constant(42)


Для определения переменной используйте функцию tf.Variable():

1
2
3
4
import tensorflow as tf

# Определение переменной
my_variable = tf.Variable(0)


Обратите внимание, что переменные должны быть инициализированы перед использованием. Для этого можно использовать функцию tf.global_variables_initializer():

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
import tensorflow as tf

# Определение переменной
my_variable = tf.Variable(0)

# Инициализация переменной
init = tf.global_variables_initializer()

with tf.Session() as sess:
    sess.run(init)
    # Теперь можно использовать переменную
    print(sess.run(my_variable))


Также, в TensorFlow есть специальный тип переменных tf.get_variable(), который позволяет определять переменные с более продвинутыми возможностями, такими как разделение переменных между устройствами и модулями.