Added distribution method nodes that take 2d tensor as input

modules/pytorch_wrapper/ptdm_cdf_tensor.py

@@ -0,0 +1,49 @@
+from typing import Any, Dict
+import torch
+
+
+class PtdmCdfTensor:
+    """
+    Computes the cumulative distribution function for the input distribution. This nodes accepts a tensor so it can be used to compute cdf for multiple values contained in a tensor.
+
+    **Note**  
+    Cdf is not supported for all distributions in PyTorch.
+
+    Args:
+            distribution (torch.distributions.distribution.Distribution): Distribution.
+            tens (torch.Tensor): Value(s) in Tensor.
+
+    category: PyTorch wrapper - Distribution
+    """
+
+    @classmethod
+    def INPUT_TYPES(cls) -> Dict[str, Any]:
+        """
+        Defines the input types for the function.
+
+        Returns:
+            Dict[str, Any]: A dictionary specifying the required input types.
+        """
+        return {
+            "required": {
+                "distribution": ("PTDISTRIBUTION", {}),
+                "tens": ("TENSOR", {})
+            }
+        }
+
+    RETURN_TYPES: tuple = ("TENSOR",)
+    FUNCTION: str = "f"
+    CATEGORY: str = "Distribution"
+
+    def f(self, distribution: torch.distributions.Distribution, tens: torch.Tensor) -> tuple:
+        """
+        Computes the cumulative distribution function for the input distribution.
+
+        Args:
+            distribution (torch.distributions.Distribution): Distribution.
+            tens (torch.Tensor): Value(s) in Tensor.
+
+        Returns:
+            tuple containing the probability
+        """
+        return (distribution.cdf(tens),)