distance_loss¶
- 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) 的張量，其值等於
兩者之間的距離損失。
