TensorFlow Operations
- Add
- Subtract
- Multiply
- Divide
- Square
- Reshape
Tensor Addition
You can add two tensors using tensorA.add(tensorB):
Example
const tensorA = tf.tensor([[1, 2], [3, 4], [5, 6]]);
const tensorB = tf.tensor([[1,-1], [2,-2], [3,-3]]);
// Tensor Addition
const tensorNew = tensorA.add(tensorB);
// Result: [ [2, 1], [5, 2], [8, 3] ]
Tensor Subtraction
You can subtract two tensors using tensorA.sub(tensorB):
Example
const tensorA = tf.tensor([[1, 2], [3, 4], [5, 6]]);
const tensorB = tf.tensor([[1,-1], [2,-2], [3,-3]]);
// Tensor Multiplication
const tensorNew = tensorA.sub(tensorB);
// Result: [ [0, 3], [1, 6], [2, 9] ]
Tensor Multiplication
You can multiply two tensors using tensorA.mul(tensorB):
Example
const tensorA = tf.tensor([1, 2, 3, 4]);
const tensorB = tf.tensor([4, 4, 2, 2]);
// Tensor Subtraction
const tensorNew = tensorA.mul(tensorB);
// Result: [ 4, 8, 6, 8 ]
Tensor Division
You can divide two tensors using tensorA.div(tensorB):
Example
const tensorA = tf.tensor([[1, 2], [3, 4], [5, 6]]);
const tensorB = tf.tensor([[1,-1], [2,-2], [3,-3]]);
// Tensor Subtraction
const tensorNew = tensorA.div(tensorB);
// Result: [ 2, 2, 3, 4 ]
Tensor Square
You can square a tensor using tensor.square():
Example
const tensorA = tf.tensor([1, 2, 3, 4]);
// Tensor Square
const tensorNew = tensorA.square();
// Result [ 1, 4, 9, 16 ]
Tensor Reshape
The number of elements in a tensor is the product of the sizes in the shape.
Since there can be different shapes with the same size, it is often useful to reshape a tensor to other shapes with the same size.
You can reschape a tensor using tensor.reshape():
Example
const tensorA = tf.tensor([[1, 2], [3, 4]]);
const tensorB = tensorA.reshape([4, 1]);
// Result: [ [1], [2], [3], [4] ]