我们可以使用input.shape来检索张量的大小。
const tensor_s = tf.tensor([2,2]).shape;
这里的形状为[2]。我们还可以创建具有特定大小的张量。...例如,下面我们创建一个形状为[2,2]的零值张量。
const input = tf.zeros([2,2]);
操作符
为了使用张量,我们需要在它们上创建操作符。...例如,要评估我们使用的张量的二次幂
const x = tf.tensor([1,2,3]);
const x2 = x.square().square();
x2张量的值为[1,16,81]。...张量释放
通常我们会生成大量的中间张量。例如,在前一个示例中,评估x2之后,我们不需要x的值。...[0, 1],[1, 0]]
因此,我们将1d张量标签转换为形状为[BATCH_SIZE,NUM_CLASSES]的张量。