2.2 或更高版本是否支持带有 float64 的 TPU?我正在使用需要 float64 才能实现数值稳定性的 VGP。我收到以下错误:
类型错误:传递给参数“输入”的值的数据类型 float64 不在允许值列表中:bfloat16、float32、int32、uint32
当前的 TF 不基于文档
2.2 或更高版本是否支持带有 float64 的 TPU?我正在使用需要 float64 才能实现数值稳定性的 VGP。我收到以下错误:
类型错误:传递给参数“输入”的值的数据类型 float64 不在允许值列表中:bfloat16、float32、int32、uint32
当前的 TF 不基于文档