TensorFlow数据类型_tensorflow数值特征用float还是int

作者：繁依Fanyi0 | 2024-02-16 18:19:35

踩

tensorflow数值特征用float还是int

一、数据类型

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

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

将数据转换成numpy格式。

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

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

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

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/96023