- 博客(1)
- 收藏
- 关注
原创 Python函数
torch.gather()函数用于根据指定索引从出入的张量中收集元素。index:一个包含了从input中收集元素的索引的。dim:指定收集操作沿着哪个维度进行。out:可选参数,指定输出张量的位置。input:输入张量。
2023-11-23 18:13:45
642
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人