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

问题描述 投票:0回答:1

此代码来自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。

tensorflow debugging tensorflow.js
1个回答
0
投票

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]]
© www.soinside.com 2019 - 2024. All rights reserved.