2020年6月29日
pytorch 常用函数
摘要： 通过索引赋值 a = torch.zeros([5,5]) index = (torch.LongTensor([0,1]),torch.LongTensor([1,2])) a.index_put_((index), torch.Tensor([1,1])) a[index] = torch.Te 阅读全文
posted @ 2020-06-29 10:12 无左无右 阅读(735) 评论(0) 推荐(0) 编辑