已解決430363個問題，去搜搜看，總會有你想問的

張量變換。

首頁猿問張量變換。

張量變換。

Python

慕婉清6462132 2021-04-09 22:15:13

我試圖用來tf.gather_nd(params, indices, name=None)從特征圖張量中檢索元素反正有沒有將這個張量[[0,2,2]]轉換[[0,0],[1,2],[2,2]] 為我需要將其用作函數中的索引我只有 [[0,2,2]]應該是這個結構indices = [[0,0],[1,2],[2,2]]params = [['3', '1','2','-1'], ['0.3', '1.4','5','0'],['5', '6','7','8']]t=tf.gather_nd(params, indices, name=None)with tf.Session() as sess: sess.run(tf.initialize_all_variables()) print(sess.run(t)) # outputs 3 5 7

查看完整描述

1 回答

慕的地6264312

TA貢獻1817條經驗獲得超6個贊

假設您嘗試將張量t0 = [[x0, x1, x2, ... xn]]轉換為張量[[0, x0], [1, x1], [2, x2], ..., [n, xn]]，可以將其與范圍張量連接起來，如下所示：

t0 = ...

N = tf.shape(t0)[1] # number of indices

t0 = tf.concat([tf.range(N), t0], 0) # [[0, 1, 2], [0, 2, 2]]

indices = tf.transpose(t0) # [[0, 0], [1, 2], [2, 2]]

這應該為您提供所需的索引。

反對回復 2021-04-20

1 回答
0 關注
197 瀏覽

關注

添加回答

舉報

0/150

提交

取消

亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

熱搜

最近搜索清空

張量變換。

張量變換。

1 回答

添加回答

張量變換。