2022年3月10日
pytorch方法整理--gather&squeeze&unsqueeze&其他一些函数
摘要： gather pytorch中gather源码形式：torch.gather(input, dim, index, *, sparse_grad = False, out = None) 然后在pytorch官方文档中，写了这样的一个例子，这个例子是三维的 out[i][j][k] = input[ 阅读全文
posted @ 2022-03-10 10:18 CharlesLC 阅读(179) 评论(0) 推荐(0)