为什么PyTorch聚集函数要求索引参数为LongTensor类型?

2024-09-30 20:20:30 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在用PyTorch编写一些代码,我遇到了聚集函数。查看文档,我看到索引参数包含一个长传感器,这是为什么?为什么它需要一个长传感器而不是像IntTensor这样的另一种类型?有什么好处


Tags: 函数代码文档类型参数传感器pytorchinttensor