服务器之家

服务器之家 > 正文

TensorFlow人工智能学习数据类型信息及转换

时间:2022-02-28 00:03     来源/作者:Swayzzu

一、数据类型

在tf中,数据类型有整型(默认是int32),浮点型(默认是float32),以及布尔型,字符串。

TensorFlow人工智能学习数据类型信息及转换

TensorFlow人工智能学习数据类型信息及转换

二、数据类型信息

 

①.device

查看tensor在哪(CPU上面或者GPU上面),可以通过.cpu(),.gpu()进行转换,如果数据所在的处理器位置不一样,则不能进行计算。

②.numpy()

将数据转换成numpy格式。

③.shape / .ndim

查看形状,.ndim查看维度,.is_tensor查看是不是tensor类型。

TensorFlow人工智能学习数据类型信息及转换

TensorFlow人工智能学习数据类型信息及转换

三、数据类型转换

 

①tf.convert_to_tensor

将数据转换成tensor类型,当从Numpy转换成tensor的时候,会默认是int64,需要指定一下类型,才能成为tf默认的类型也就是int32。

②tf.cast()

可以实现tensor的数据类型转换。

TensorFlow人工智能学习数据类型信息及转换

TensorFlow人工智能学习数据类型信息及转换

TensorFlow人工智能学习数据类型信息及转换

注意:在深度学习中,是需要对参数求梯度的,需要variable包装一下,就拥有了trainable属性,这样就才求梯度。假如是自己写传播过程,更新后的参数也需要用variable包装。

TensorFlow人工智能学习数据类型信息及转换

以上就是TensorFlow人工智能学习数据类型信息及转换的详细内容,更多关于TensorFlow数据类型的资料请关注服务器之家其它相关文章!

原文链接:https://blog.csdn.net/Swayzzu/article/details/121057787

相关文章

热门资讯

2022年最旺的微信头像大全 微信头像2022年最新版图片
2022年最旺的微信头像大全 微信头像2022年最新版图片 2022-01-10
蜘蛛侠3英雄无归3正片免费播放 蜘蛛侠3在线观看免费高清完整
蜘蛛侠3英雄无归3正片免费播放 蜘蛛侠3在线观看免费高清完整 2021-08-24
背刺什么意思 网络词语背刺是什么梗
背刺什么意思 网络词语背刺是什么梗 2020-05-22
yue是什么意思 网络流行语yue了是什么梗
yue是什么意思 网络流行语yue了是什么梗 2020-10-11
2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全
2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全 2019-12-26
返回顶部