我有一个数组A
矩阵(或一个三维张量),我想做以下事情:
用数字表示每个矩阵,因此A
为[1,2,3,4,...,]
,并且让我们说我们有一个长度为3的窗口,我想将其作为输入传递给TensorFlow图形4-dim数组[[1,2,3],[2,3,4],[3,4,5],....]
。这样做最有效的方法是什么? (它有点像带有常量内核的卷积,但没有对得到的矩阵进行求和)。
目前这就是我正在做的事情:
input_NN = [data[t, t + window] for t in range(my_range)]
然后我将它传递给TF占位符。
我想到在numpy
中更好的方法,并将结果传递给占位符,或者通过直接传递A
在TensorFlow中有快速的方法吗?