哪个函数用于获取python中tensorflow中张量的数据类型?我需要根据不同的张量定义动态数据类型.
解决方法:
您可以使用x.dtype获取类型,如下所示:
import tensorflow as tf
x=tf.constant([1,2])
x.dtype
这会打印tf.int32
哪个函数用于获取python中tensorflow中张量的数据类型?我需要根据不同的张量定义动态数据类型.
解决方法:
您可以使用x.dtype获取类型,如下所示:
import tensorflow as tf
x=tf.constant([1,2])
x.dtype
这会打印tf.int32