在mxnet中高效创建压缩带状对角矩阵

作者: v-star*위위
发布时间: 2024-01-14 05:50:20 (7月前)
转自：

2 条回复

0#
回复此人
部落用户 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 如果在你的手术中， <code> k </code> 和 <code> n </code> 是不变的，你可以用你的组合做你想做的事 <a href="https://mxnet.apache.org/api/python/ndarray/ndarray.html#mxnet.ndarray.gather_nd" rel="nofollow noreferrer"> <code> mxnet.nd.gather_nd() </code> </A> 和 <a href="https://mxnet.apache.org/api/python/ndarray/ndarray.html#mxnet.ndarray.gather_nd" rel="nofollow noreferrer"> <code> mx.nd.scatter_nd </code> </A> 。即使生成索引张量也同样效率低，因为您只需要执行一次，这不会有问题。你会想要使用 <code> gather_nd </code> 从原始数组中有效地“复制”您的数据，然后使用 <code> scatter_nd </code> 将它们分散到最终的矩阵形状。或者，您可以简单地连接一个 <code> 0 </code> 元素到您的输入数组（例如 <code> [a_1, a_2, a_3] </code> 会变成 <code> [0, a_1, a_2, a_3] </code> ）然后只使用 <code> mxnet.nd.gather_nd() </code> 将元素复制到最终矩阵中。 </p> </DIV>

编辑

登录后才能参与评论