Tensor Operations

1/**2 * Example 02: Tensor Operations3 *4 * Explore arithmetic, mathematical, and reduction operations on tensors.5 * Deepbox supports 90+ tensor operations with full broadcasting.6 */78import {9  add,10  cos,11  div,12  exp,13  log,14  max,15  mean,16  min,17  mul,18  sin,19  sqrt,20  sub,21  sum,22  tensor,23} from "deepbox/ndarray";2425console.log("=== Tensor Operations ===\n");2627// Basic arithmetic28const a = tensor([1, 2, 3, 4]);29const b = tensor([5, 6, 7, 8]);3031console.log("a =", a.toString());32console.log("b =", b.toString());3334console.log("\nArithmetic Operations:");35console.log("a + b =", add(a, b).toString());36console.log("a * b =", mul(a, b).toString());37console.log("a - b =", sub(a, b).toString());38console.log("a / b =", div(a, b).toString());3940// Mathematical functions41const x = tensor([1, 4, 9, 16]);42console.log("\nMathematical Functions:");43console.log("x =", x.toString());44console.log("sqrt(x) =", sqrt(x).toString());45console.log("exp(x) =", exp(tensor([0, 1, 2])).toString());46console.log("log(x) =", log(x).toString());4748// Trigonometric functions49const angles = tensor([0, Math.PI / 4, Math.PI / 2, Math.PI]);50console.log("\nTrigonometric Functions:");51console.log("angles =", angles.toString());52console.log("sin(angles) =", sin(angles).toString());53console.log("cos(angles) =", cos(angles).toString());5455// Reduction operations56const matrix = tensor([57  [1, 2, 3],58  [4, 5, 6],59]);6061console.log("\nReduction Operations:");62console.log("matrix =");63console.log(matrix.toString());64console.log("sum(matrix) =", sum(matrix).toString());65console.log("mean(matrix) =", mean(matrix).toString());66console.log("max(matrix) =", max(matrix).toString());67console.log("min(matrix) =", min(matrix).toString());6869// Axis-wise reductions70console.log("\nAxis-wise Reductions:");71console.log("sum(matrix, axis=0) =", sum(matrix, 0).toString());72console.log("sum(matrix, axis=1) =", sum(matrix, 1).toString());73console.log("mean(matrix, axis=0) =", mean(matrix, 0).toString());7475console.log("\n✓ Tensor operations complete!");76