2022年4月18日
Torch.gather
摘要： Torch.gather 函数定义： torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor 参数： input (Tensor) – 源tensor dim (int)) –索引轴 index (LongTe 阅读全文
posted @ 2022-04-18 10:28 LibraXiong 阅读(150) 评论(0) 推荐(1)