如何在 Tensorflow.js 中按给定的整数比例因子放大张量的大小？

Question

我知道可以在普通JS中实现缩放函数，但如果我有一个这样的张量:

a = tf.tensor([[1,2],[3,4]])
a.print();
/*
Tensor
[[1, 2],
[3, 4]]
*/

Tensorflow.js是否有一种原生的方式来缩放张量，而不需要首先将其转换为香草JS数据结构?例如，如果我把上面的张量在所有维度上放大2，我会得到这个:

Tensor
[[1, 1, 2, 2],
[1, 1, 2, 2],
[3, 3, 4, 4],
[3, 3, 4, 4]]

我怎么能缩放Tensorflow.js张量没有首先转换成一个香草JS数据结构?

Answer 1

mirrorPad就会这样做

x = tf.tensor([[1,2],[3,4]])
x.mirrorPad([[1, 1], [1, 1]], 'symmetric').print();