Función torch.max de PyTorch

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

torch.maxEs una función en PyTorch que se utiliza para calcular el valor máximo de los tensores.

Definición de la función

torch.max(input, dim, keepdim, out)

Ejemplo de uso

Ejemplo

import torch

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

# máximo global
print("máximo global:", torch.max(x))

# máximo a lo largo de la dimensión
print("dim=0 máximo:", torch.max(x, dim=0))

El resultado de salida es:

全局最大: tensor(6)
dim=0 最大: torch.return_types.max(values=tensor([4, 5, 6]), indices=tensor([1, 1, 1]))

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

Otras extensiones