如何在TensorFlow中获取张量的数据类型?

7
哪个函数用于在python中获取tensorflow张量的数据类型?我需要根据不同的张量定义动态数据类型。
2个回答

12

您可以使用x.dtype获取类型,如下所示:

import tensorflow as tf
x=tf.constant([1,2])
x.dtype

这将打印tf.int32


0
如果张量是Pandas数据框,则使用以下代码:
print(df.dtypes)

针对特定列,请使用

print (df['Prices'].dtypes)

参考资料:链接


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接