距离损失¶
- class torchrl.objectives.distance_loss(v1: Tensor, v2: Tensor, loss_function: str, strict_shape: bool = True)[源代码]¶
计算两个张量之间的距离损失。
- 参数：:
v1 (Tensor) – 与 v2 形状兼容的张量
v2 (Tensor) – 与 v1 形状兼容的张量
loss_function (str) – “l2”、“l1” 或 “smooth_l1” 之一，表示要使用的损失函数。
strict_shape (bool) – 如果为 False，则允许 v1 和 v2 具有不同的形状。默认值为
True。
- 返回值：:
形状为 v1.view_as(v2) 或 v2.view_as(v1) 的张量，其值等于两个之间的距离损失
两个。
