如何解释"tff.to_type((tf.int64，[2])“？

Question

我正在学习Tensorflow-Federated，而我不知道如何解释语法tff.to_type((tf.int64, [2]))。我认为tff.to_type可以创建一个tff数据类型，但我不知道2的含义。

Answer 1

因此，tff.to_type创建一个具有dtype tf.int64和shape [2]的tff类型。这就像定义一个描述张量属性的tf.TensorSpec。它有一个shape参数和一个dtype参数。为了更好地理解，请运行：

import tensorflow as tf
tf.TensorSpec(dtype=tf.int64, shape=[2])

TensorSpec(shape=(2,), dtype=tf.int64, name=None)

因此，这个张量将有两个数据类型为tf.int64的值。