Replaces nn.functional.tanh with torch.tanh

pfnet · Sep 4, 2020 · 08e23a4 · 08e23a4
1 parent e6c9b01
commit 08e23a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pfrl/functions/bound_by_tanh.py b/pfrl/functions/bound_by_tanh.py
@@ -20,4 +20,4 @@ def bound_by_tanh(x, low, high):
     high = torch.as_tensor(high, dtype=x.dtype, device=x.device)
     scale = (high - low) / 2
     loc = (high + low) / 2
-    return nn.functional.tanh(x) * scale + loc
+    return torch.tanh(x) * scale + loc