TensorFlow 操作
- 添加
- 减去
- 乘
- 划分
- 正方形
- 重塑
张量添加
您可以使用以下方式添加两个张量 张量A.添加(张量B):
例子
const tensorA = tf.tensor([[1, 2], [3, 4], [5, 6]]);
const tensorB = tf.tensor([[1,-1], [2,-2], [3,-3]]);
// 张量添加
const tensorNew = tensorA.add(tensorB);
// 结果: [ [2, 1], [5, 2], [8, 3] ]
张量减法
您可以使用以下方法减去两个张量 张量A.sub(张量B):
例子
const tensorA = tf.tensor([[1, 2], [3, 4], [5, 6]]);
const tensorB = tf.tensor([[1,-1], [2,-2], [3,-3]]);
// 张量减法
const tensorNew = tensorA.sub(tensorB);
// 结果: [ [0, 3], [1, 6], [2, 9] ]
张量乘法
您可以使用以下方法将两个张量相乘 张量A.乘以(张量B):
例子
const tensorA = tf.tensor([1, 2, 3, 4]);
const tensorB = tf.tensor([4, 4, 2, 2]);
// 张量乘法
const tensorNew = tensorA.mul(tensorB);
// 结果: [ 4, 8, 6, 8 ]
张量分部
您可以使用以下方法划分两个张量 张量A.div(张量B):
例子
const tensorA = tf.tensor([2, 4, 6, 8]);
const tensorB = tf.tensor([1, 2, 2, 2]);
// 张量划分
const tensorNew = tensorA.div(tensorB);
// 结果: [ 2, 2, 3, 4 ]
张量平方
你可以使用以下方法对张量进行平方 tensor.square():
例子
const tensorA = tf.tensor([1, 2, 3, 4]);
// 张量平方
const tensorNew = tensorA.square();
// 结果 [ 1, 4, 9, 16 ]
张量重塑
张量中元素的数量是形状大小的乘积。
由于相同大小可以有不同的形状,因此将张量重塑为具有相同大小的其他形状通常很有用。
您可以使用以下方式重塑张量 tensor.reshape():
例子
const tensorA = tf.tensor([[1, 2], [3, 4]]);
const tensorB = tensorA.reshape([4, 1]);
// 结果: [ [1], [2], [3], [4] ]