Shortcuts

torcheval.metrics.Min

class torcheval.metrics.Min(*, device: device | None = None)

计算所有输入张量中所有元素的最小值。 其功能版本是 torch.min(input)

示例:

>>> import torch
>>> from torcheval.metrics import Min
>>> metric = Min()
>>> metric.update(torch.tensor([[1, 2], [3, 4]]))
>>> metric.compute()
tensor(1.)

>>> metric.update(torch.tensor(-1)).compute()
tensor(-1.)

>>> metric.reset()
>>> metric.update(torch.tensor(5)).compute()
tensor(5.)
__init__(*, device: device | None = None) None

初始化一个度量对象及其内部状态。

使用 self._add_state() 来初始化你的度量类的状态变量。 状态变量应该是 torch.Tensor,一个 torch.Tensor 的列表,一个以 torch.Tensor 为值的字典, 或者一个 torch.Tensor 的双端队列。

方法

__init__(*[, device])

初始化一个度量对象及其内部状态。

compute()

实现此方法以从状态变量计算并返回最终指标值。

load_state_dict(state_dict[, strict])

从state_dict加载度量状态变量。

merge_state(metrics)

实现此方法以将当前度量的状态变量更新为当前度量和输入度量的合并状态。

reset()

将度量状态变量重置为其默认值。

state_dict()

将度量状态变量保存在state_dict中。

to(device, *args, **kwargs)

将度量状态变量中的张量移动到设备。

update(input)

实现此方法以更新您的指标类的状态变量。

属性

device

Metric.to()的最后一个输入设备。