From 392519634c7e352c6907e91825bacb6ca042e441 Mon Sep 17 00:00:00 2001
From: mei1127 <shanxing.mei@intel.com>
Date: Wed, 10 Jan 2024 09:35:53 +0800
Subject: [PATCH] revised constant.js

---
 src/constant.js | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/constant.js b/src/constant.js
index d34fb4b..902d330 100644
--- a/src/constant.js
+++ b/src/constant.js
@@ -2,21 +2,21 @@
 
 import {cast} from './cast.js';
 import {Tensor, sizeOfShape} from '../src/lib/tensor.js';
-
 /**
  * Create a constant array of specified data type and shape,
  * which contains data incrementing by step.
+ * @param {Number} start
+ * @param {Number} step
  * @param {Array} outputShape
+ * @param {string} type
  * @return {Tensor}
  */
-
 export function constant(start, step, outputShape, type = 'float32') {
   const outputElementCount = sizeOfShape(outputShape);
-  const resultArray = [];
+  const data = [];
   for (let i = 0; i < outputElementCount; i++) {
-    resultArray.push(start + i * step);
+    data.push(start + i * step);
   }
-  const resultToTensor = new Tensor(outputShape, resultArray);
-  const output = cast(resultToTensor, type);
-  return output;
+  const tensor = new Tensor(outputShape, data);
+  return cast(tensor, type);
 }