[Erro] Backpropagating indexed elements from Variable

autograd
4

Hi,

Would

mytriu = Variable(torch.triu(torch.randn(5,5)))
idx = torch.nonzero(torch.triu(torch.ones(*mytriu.size())).view(-1)).view(-1)
mytriu_as_vec = vec = mytriu.view(-1)[idx]

work for you?

Best regards

Thomas