Какая операция с тензорами a, b:
a = torch.ones([5,4])
b = torch.Tensor([[1,2,3,4], [5,6,7,8], [9,10,11,12], [13,14,15,16], [17,18,19,20]])
позволяет получить следующий вывод?
tensor([2., 3., 4., 5.])