如何给tensor变量赋值

最新推荐文章于 2025-02-02 21:37:03 发布

原创最新推荐文章于 2025-02-02 21:37:03 发布 · 1.1w 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow

tensorflow学习专栏收录该内容

5 篇文章

订阅专栏

本文介绍了使用TensorFlow进行变量赋值的三种方法，包括直接初始化、使用assign方法以及通过会话加载值，展示了如何在TensorFlow环境中灵活操作变量。

部署运行你感兴趣的模型镜像

以下有三种方式可以给tensor变量赋值，代码如下：

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import numpy as np
import os 
import sys
import tensorflow as tf

workpath=os.path.dirname(sys.argv[0])
os.chdir(workpath)          #指定py文件执行路径为当前工作路径

def main():
    
   x = tf.Variable(1)
   x2=tf.assign(x,2)
   x3=x.assign(3)
   with tf.Session() as sess:
      print(sess.run(x2))
      print(sess.run(x3))
      x.load(4, sess)
      x4=x
      print(sess.run(x4))

if __name__=="__main__":
    main()

执行结果如图：
在这里插入图片描述