我正在学习Tensorflow Federated，但我不知道如何解释语法tff.to_type((tf.int64, [2])).我认为tff.to_type可以创建tff数据类型，但我不知道[2]的含义.
我正在学习Tensorflow Federated，但我不知道如何解释语法tff.to_type((tf.int64, [2])).我认为tff.to_type可以创建tff数据类型，但我不知道[2]的含义.
因此，tff.to_type创建了具有数据类型tf.int64和形状[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的值.