TensorFlow 操作

添加
减去
乘
划分
正方形
重塑

张量加法

您可以使用添加两个张量tensorA.add(tensorB):

示例

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

亲自试一试 »

张量减法

您可以使用减去两个张量tensorA.sub(tensorB):

示例

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

亲自试一试 »

张量乘法

您可以使用两个张量相乘tensorA.mul(tensorB):

示例

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 ]

亲自试一试 »

张量除法

您可以使用以下方法除以两个张量tensorA.div(tensorB):

示例

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

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

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

亲自试一试 »

张量平方

您可以使用以下方法对张量求平方tensor.square() :

示例

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

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

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

亲自试一试 »

张量重塑

张量中元素的数量是形状大小的乘积。

由于可以有具有相同大小的不同形状，因此将张量重塑为具有相同大小的其他形状通常很有用。

您可以使用重塑张量tensor.reshape() :

示例

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

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

亲自试一试 »