torch.aminmax

torch.aminmax(input, *, dim=None, keepdim=False, out=None) -> (Tensor min, Tensor max)

計算 input 張量的最小值和最大值。

參數

input (Tensor) – 輸入張量

關鍵字參數

• dim (選用[int]) – 計算值的維度。 如果 None，則計算整個 input 張量的值。 預設值為 None。

• keepdim (bool) – 如果 True，則縮減的維度將保留在輸出張量中，作為大小為 1 的維度以進行廣播；否則，它們將被移除，如同調用 (torch.squeeze())。 預設值為 False。

• out (選用[Tuple[Tensor, Tensor]]) – 可選的張量，用於寫入結果。 必須具有與預期輸出相同的形狀和 dtype。 預設值為 None。

返回

一個名為 (min, max) 的具名元組，包含最小值和最大值。

引發

RuntimeError – 如果計算值的任何維度的大小為 0。

注意

如果至少有一個值為 NaN，則 NaN 值將傳播到輸出。

另請參閱

torch.amin() 僅計算最小值，torch.amax() 僅計算最大值

範例

>>> torch.aminmax(torch.tensor([1, -3, 5]))
torch.return_types.aminmax(
min=tensor(-3),
max=tensor(5))

>>> # aminmax propagates NaNs
>>> torch.aminmax(torch.tensor([1, -3, 5, torch.nan]))
torch.return_types.aminmax(
min=tensor(nan),
max=tensor(nan))

>>> t = torch.arange(10).view(2, 5)
>>> t
tensor([[0, 1, 2, 3, 4],
        [5, 6, 7, 8, 9]])
>>> t.aminmax(dim=0, keepdim=True)
torch.return_types.aminmax(
min=tensor([[0, 1, 2, 3, 4]]),
max=tensor([[5, 6, 7, 8, 9]]))