torch常用操作总结
torch.unsqueeze(1) torch.unsqueeze(0) torch.view() arr[:, :, np.newaxis] 复制
torch.unsqueeze(1) torch.unsqueeze(0) torch.view() arr[:, :, np.newaxis]
compute the avage pos: torch_scatter scatter(src,index.dim,reduce="means")(sum,mul,min,max,sum)
torch.min include values and indices 复制
torch.min include values and indices