PyTorch torch.min Function


Pytorch torch 参考手册Pytorch torch Reference Manual

torch.minIt is a function in PyTorch used to compute the minimum value of tensors.

Function Definition

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

Usage Example

Example

import torch

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

print(Global minimum:, torch.min(x))
print(dim=0 minimum:, torch.min(x, dim=0))

Pytorch torch 参考手册Pytorch torch Reference Manual

Other Extensions