PyTorch torch.amin 函数
Pytorch torch 参考手册torch.amin 是 PyTorch 中用于返回张量沿指定维度的最小值的函数。
函数定义
torch.amin(input, dim, keepdim=False)
使用示例
实例
import torch
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
# 返回所有元素的最小值
print("全局最小:", torch.amin(x))
# 沿 dim=0 最小值
print("dim=0 最小:", torch.amin(x, dim=0))
# 沿 dim=1 最小值
print("dim=1 最小:", torch.amin(x, dim=1))
x = torch.tensor([[1, 3, 2], [4, 1, 3]])
# 返回所有元素的最小值
print("全局最小:", torch.amin(x))
# 沿 dim=0 最小值
print("dim=0 最小:", torch.amin(x, dim=0))
# 沿 dim=1 最小值
print("dim=1 最小:", torch.amin(x, dim=1))
输出结果为：
全局最小: tensor(1) dim=0 最小: tensor([1, 1, 2]) dim=1 最小: tensor([1, 1])