qml.pytrees

警告

除非您是PennyLane或插件开发者，否则您可能不需要使用这些函数。

一个用于处理pytrees的内部模块。

函数

flatten(obj[, is_leaf])	将一个pytree展平为叶子和结构。
is_pytree(type_)	如果 type_ 是一个注册的 Pytree，则返回 True。
register_pytree(pytree_type, flatten_fn, ...)	注册一个带有所有可用 pytree 后端的类型。
unflatten(数据, 结构)	将数据绑定到结构以重建pytree对象。

类

PyTreeStructure([类型_, 元数据, 子节点])

一个pytree数据结构，包含类型、元数据和子pytree结构。

变量

leaf	一个pytree数据结构，包含类型、元数据和子pytree结构。

代码/qml_pytrees

 

下载 Python 脚本

 

下载笔记本

 

在GitHub上查看