python - tf.create_partitioned_variables 如何工作？

Question

我正在尝试弄清楚如何使用tf.create_partitioned_variables我正在阅读文档，但我很难理解。

谁能解释一下它是如何工作的并给出一些用法示例？

据我了解，我可以使用它从变量中获取切片列表。我只是不明白如何获得切片

例如:我如何从 tf.Variable(np.array([[1.0],[3.0]]), dtype= 获取 [[1.],[3.]] 列表tf.float32)

或列表

[[[1 0] [3 0]], [[0 5] [0 7]]]

来自

[[[1 0]
  [3 0]]

 [[0 5]
  [0 7]]]

Answer 1

前 3 个参数是必需的。第一个是输入张量的形状。二是拆分规范。该 API 目前仅支持沿一维拆分。分割规范与形状具有相同的维数，其中一个分割 >= 1，其他分割为 1。最后一个参数是张量本身，或者返回它的可调用对象。

第一个例子:

tf.create_partitioned_variables(v.shape, [2, 1], v)

第二个例子:

[tf.squeeze(v) 
    for v in tf.create_partitioned_variables(
        v.shape, [2, 1, 1], v)]