Fatal编程技术网
  • pytorch/
  • Pytorch 从屏蔽索引中获取张量列表

Pytorch 从屏蔽索引中获取张量列表

pytorch

Pytorch 从屏蔽索引中获取张量列表,pytorch,tensor,Pytorch,Tensor,我想得到一个基于不同组的张量列表 e、 g 我希望我的结果是正确的 res= [[0.3018, -0.0079], [1.4995, -1.4422], [1.6007]] 我想知道如何才能达到这个结果，我选中了收集和索引选择， 但是我不能得到像上面那样的结果 谢谢大家! 怎么样 res=[x[index==i_u]用于索引中的i_u.unique（） 看起来您可以为循环执行。您期望的输出方式无论如何都不允许矢量化。 res= [[0.3018, -0.0079], [1.4995, -1.

我想得到一个基于不同组的张量列表

e、 g

我希望我的结果是正确的

res= [[0.3018, -0.0079], [1.4995, -1.4422], [1.6007]]
我想知道如何才能达到这个结果，我选中了
收集
和
索引选择
， 但是我不能得到像上面那样的结果

谢谢大家!

怎么样

res=[x[index==i_u]用于索引中的i_u.unique（）
看起来您可以为循环执行
。您期望的输出方式无论如何都不允许矢量化。

res= [[0.3018, -0.0079], [1.4995, -1.4422], [1.6007]]