Sub-Store节点转换是将较大的模型训练图划分为多个较小的子图,每个子图独立训练或优化。以下是实现Sub-Store节点转换的清晰步骤
定义转换函数:
创建一个Python函数,接受输入张量,返回Sub-Store节点。
该函数将输入张量转换为Sub-Store节点结构,可能通过遍历输出节点的结构,将其分解为多个子图节点。
使用Sub-Store构造器:
将输入张量传递给转换函数,得到Sub-Store节点。
使用Sub-Store的构造器,将输入张量转化为Sub-Store节点结构,如多个子图节点。
整合节点到模型中:
将转换后的Sub-...