Tensorflow.js:tf.pad导致TypeError:t.map不是一个函数

时间:2019-12-09 08:03:25

标签: tensorflow debugging tensorflow.js

此代码来自TF API文档:

let t = tf.tensor([[1, 2, 3], [4, 5, 6]])
let padding = tf.tensor([[1, 1,], [2, 2]])

执行时:

tf.pad(t, padding, "CONSTANT")

我得到:

  

TypeError:t.map不是函数

我正在使用最新版本的tfjs。

1 个答案:

答案 0 :(得分:1)

padding是一个普通的js元组数组(arrray数组),而不是张量。

就目前而言,版本1.3.1仅支持CONSTANT模式。这是要走的路:

let t = tf.tensor([[1, 2, 3], [4, 5, 6]])
let padding = [[2, 2,], [1, 1]]

tf.pad(t, padding).print()
//  [[0, 0, 0, 0, 0],
//   [0, 0, 0, 0, 0],
//   [0, 1, 2, 3, 0],
//   [0, 4, 5, 6, 0],
//   [0, 0, 0, 0, 0],
//   [0, 0, 0, 0, 0]]