다음 형식의 텐서가 있다고 가정합니다.
[[-5, 0, -1],
[3, 100, 87],
[17, -34, 2],
[45, 1, 25]]
각 행에서 최대값을 찾고 다음과 같이 순위 1 텐서를 반환하고 싶습니다.
[0,
100,
17,
45]
PyTorch에서 어떻게 합니까?
torch.max()
기능을 사용할 수 있습니다 . 그래서 당신은 다음과 같은 것을 할 수 있습니다
x = torch.Tensor([[-5, 0, -1],
[3, 100, 87],
[17, -34, 2],
[45, 1, 25]])
out, inds = torch.max(x,dim=1)
그러면 각 행(차원 1)의 최대값이 반환됩니다. 인덱스와 함께 최대값을 반환합니다.
이 기사는 인터넷에서 수집됩니다. 재 인쇄 할 때 출처를 알려주십시오.
침해가 발생한 경우 연락 주시기 바랍니다[email protected] 삭제
몇 마디 만하겠습니다