TensorFlow提取列

Question

我有一個形狀張量（10,100,20,3）。基本上，它可以被認爲是一批圖像。因此圖像高度爲100，寬度爲20，通道深度爲3.

我運行了一些計算來生成一組10 * 50索引，對應於50列，我希望在批處理中保留每個圖像。索引存儲在形狀張量（10,50）中。我想最終形成一個張量（10,50,20,3）。

我已經看過tf.batch_nd()，但我無法弄清索引如何實際使用的語義。

有什麼想法？

Answer 1

我不能評論這個問題，因爲代表性低，所以使用了一個答案。

你能澄清一下你的問題嗎，也許有一個小的具體例子，使用非常小的張量？

你指的是什麼「列」？你說你想每個圖像保留50列（大概是50個數字）。如果是這樣，那麼（10,50）形狀看起來就像你想要的一樣 - 批次中的每個圖像都有50個數字。您提到的（10,50,20,3）形狀將爲每個「image_column x通道」分配50個數字。這是20 * 3 * 50 =每個圖像3000個數字。你如何從你擁有的50個建造它們？您可以鏈接到tf.batch_nd()。我沒有發現任何相似和相關的東西。