Función PyTorch torch.amax

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.amaxEs una función de PyTorch que se utiliza para devolver el valor máximo de un tensor a lo largo de la dimensión especificada.

Definición de la función

torch.amax(input, dim, keepdim=False)

Ejemplo de uso

Ejemplo

import torch

x = torch.tensor([[1, 3, 2], [4, 1, 3]])

# Devuelve el valor máximo de todos los elementos
print("Máximo global:", torch.amax(x))

# Valor máximo a lo largo de dim=0
print("Máximo en dim=0:", torch.amax(x, dim=0))

# Valor máximo a lo largo de dim=1
print("Máximo en dim=1:", torch.amax(x, dim=1))

El resultado es:

全局最大: tensor(4)
dim=0 最大: tensor([4, 3, 3])
dim=1 最大: tensor([3, 4])

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones