04 2024 档案

torch.gather
摘要：解释： 以下面代码为例： index = torch.tensor([[2, 1, 0]]) tensor_1 = tensor_0.gather(dim=1, index) print(tensor_1) (1) output.shape = index.shape # 确定最后输出的output 阅读全文
posted @ 2024-04-09 10:53 SXQ-BLOG 阅读(10) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示