Opérations TensorFlow

  • Ajouter
  • Soustraire
  • Multiplier
  • Diviser
  • Carré
  • Remodeler

Ajout de tenseur

Vous pouvez ajouter deux tenseurs en utilisant tensorA.add(tensorB) :

Exemple

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] ]


Soustraction du tenseur

Vous pouvez soustraire deux tenseurs en utilisant tensorA.sub(tensorB) :

Exemple

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.sub(tensorB);

// Result: [ [0, 3], [1, 6], [2, 9] ]


Multiplication du tenseur

Vous pouvez multiplier deux tenseurs en utilisant tensorA.mul(tensorB) :

Exemple

const tensorA = tf.tensor([1, 2, 3, 4]);
const tensorB = tf.tensor([4, 4, 2, 2]);

// Tensor Multiplication
const tensorNew = tensorA.mul(tensorB);

// Result: [ 4, 8, 6, 8 ]


Division du tenseur

Vous pouvez diviser deux tenseurs en utilisant tensorA.div(tensorB) :

Exemple

const tensorA = tf.tensor([[1, 2], [3, 4], [5, 6]]);
const tensorB = tf.tensor([[1,-1], [2,-2], [3,-3]]);

// Tensor Division
const tensorNew = tensorA.div(tensorB);

// Result: [ 2, 2, 3, 4 ]


Carré tenseur

Vous pouvez mettre un tenseur au carré en utilisant tensor.square() :

Exemple

const tensorA = tf.tensor([1, 2, 3, 4]);

// Tensor Square
const tensorNew = tensorA.square();

// Result [ 1, 4, 9, 16 ]


Remodelage du tenseur

Le nombre d'éléments dans un tenseur est le produit des tailles dans la forme.

Puisqu'il peut y avoir différentes formes avec la même taille, il est souvent utile de remodeler un tenseur en d'autres formes avec la même taille.

Vous pouvez remodeler un tenseur en utilisant tensor.reshape() :

Exemple

const tensorA = tf.tensor([[1, 2], [3, 4]]);
const tensorB = tensorA.reshape([4, 1]);

// Result: [ [1], [2], [3], [4] ]