Tensorflow-keras

最新推荐文章于 2023-12-30 08:02:56 发布

原创最新推荐文章于 2023-12-30 08:02:56 发布 · 289 阅读

0 ·

CC 4.0 BY-SA版权

deeplearning 专栏收录该内容

1 篇文章

订阅专栏

本文演示了如何使用Python代码检查TensorFlow及其Keras模块的安装版本和路径，同时展示了如何创建并运行一个简单的张量操作。

导入tf.keras

import tensorflow as tf 
import tensorflow.keras as keras
if __name__ == '__main__':
	print(tf.__version__)  // check tensorflow version
	print(keras.__version__)// check tensorflow  keras version
	print(tf.__path__) // check tensorflow install path
	// check whether tf is gpu or not
	a = tf.constant([1,2,3,4,5,6], shape=[2,3],name='a')
	sess = tf.Session()
	print(sess.run(a))