pytorch 用视图替换张量中的数据

问题描述

我正在尝试构建一个 TreeTensor 数据结构,它仍然是一个 pytorch 张量:它是一个二叉树,每个节点要么是一个普通的 torch.Tensor,要么是一个 TreeTensor。它的根包含其存储中的所有数据作为一个平面一维张量,所有子节点都是这个父数据的视图。

这样做时,我面临以下问题:为了能够构建我的 TreeTensor,一些节点需要从拥有数据更改为现在存储在父一维数组中的相同数据的视图。这是因为在构建它时,我需要自底向上方法的灵活性 - 我首先创建子项,然后父项指向这些子项。如何将数据移出张量并用相同维度的视图替换它?

动机:我想创建一个信号的层次结构,可以很好地使用 x.y 类型的符号进行访问。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)