From 047ce6f8ad8f0c4da8ee9b37bcd5fe29e4b950e8 Mon Sep 17 00:00:00 2001 From: HowToSD Date: Sat, 1 Mar 2025 13:00:05 -0800 Subject: [PATCH] Updated docstring for Pt Linspace. --- docs/reference/node_reference.md | 2 +- docs/reference/pt_linspace.md | 2 +- modules/pytorch_wrapper/pt_linspace.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/reference/node_reference.md b/docs/reference/node_reference.md index e0aebcc..81f60c7 100644 --- a/docs/reference/node_reference.md +++ b/docs/reference/node_reference.md @@ -243,7 +243,7 @@ | [Pt From Numpy](pt_from_numpy.md) | Converts a NumPy ndarray to a PyTorch tensor while preserving its data type. | | [Pt Full](pt_full.md) | Creates a PyTorch tensor filled with a specified value using the size entered in the text field. | | [Pt Int Create](pt_int_create.md) | Creates a PyTorch tensor with 32-bit integer | -| [Pt Linspace](pt_linspace.md) | Creates a PyTorch tensor using `torch.arange` with the specified start, end, and steps values. | +| [Pt Linspace](pt_linspace.md) | Creates a PyTorch tensor using `torch.linspace` with the specified start, end, and steps values. | | [Pt Ones](pt_ones.md) | Creates a PyTorch tensor of ones using the size entered in the text field. | | [Pt Rand](pt_rand.md) | Creates a PyTorch tensor with values sampled from a uniform distribution | | [Pt Rand Int](pt_rand_int.md) | Creates a PyTorch tensor filled with random integers within a specified range using the size entered in the text field. | diff --git a/docs/reference/pt_linspace.md b/docs/reference/pt_linspace.md index 5d3c7f0..df29f03 100644 --- a/docs/reference/pt_linspace.md +++ b/docs/reference/pt_linspace.md @@ -1,5 +1,5 @@ # Pt Linspace -Creates a PyTorch tensor using `torch.arange` with the specified start, end, and steps values. +Creates a PyTorch tensor using `torch.linspace` with the specified start, end, and steps values. The start and end values are parsed from string inputs and converted to a float. The steps value is converted to an int. diff --git a/modules/pytorch_wrapper/pt_linspace.py b/modules/pytorch_wrapper/pt_linspace.py index c7a16a8..731b1bc 100644 --- a/modules/pytorch_wrapper/pt_linspace.py +++ b/modules/pytorch_wrapper/pt_linspace.py @@ -12,7 +12,7 @@ from cda.utils import str_to_number, str_to_number_with_default class PtLinspace: """ - Creates a PyTorch tensor using `torch.arange` with the specified start, end, and steps values. + Creates a PyTorch tensor using `torch.linspace` with the specified start, end, and steps values. The start and end values are parsed from string inputs and converted to a float. The steps value is converted to an int.