TensorFlow中与torch.full相同的函数

在 TensorFlow 中，与 PyTorch 中的 `torch.full` 函数相同的函数是 `tf.fill`。该函数的语法如下：

tf.fill(dims, value, name=None)

其中，`dims` 表示输出张量的维度，`value` 表示输出张量的填充值，`name` 表示操作的名称。例如，要创建一个形状为 (2, 3)、填充值为 0 的张量，可以使用以下代码：

import tensorflow as tf

x = tf.fill(dims=(2, 3), value=0)
print(x)

输出：

tf.Tensor(
[[0 0 0]
 [0 0 0]], shape=(2, 3), dtype=int32)