2017-07-13 6 views
1

テンソルの最初の2列を抽出します。テンソルからサブマトリックスを抽出するには?

例えば

、 はこのテンソル

[[0, 1, -1, -2, -3], 

[1, 0, -8, -2, -3]] 

が出力さ

[[0,1], 
[1,0]] 

する必要があります考える私は(tf.unstackでプレイ)が、完全に成功することができませんでした。あなたがいる場合

tensor[:,0:2] 

+0

テンソルはどのような種類のオブジェクトですか? 'リスト'?あなたは 'res = [xのテンソルでx [:2]を行うことができます ' –

+0

テンソルフロー変数です。 tf.Variable(...) –

答えて

1

あなたはあなたがtensorflowのPythonのスライス演算子を使用して、それをもう少し読みやすいクリスRoofeのソリューションよりも

import tensorflow as tf 
tf.Session().run(tf.slice([[0,1,-1,-2,-3],[1,0,-9, -2, -3]], [0, 0], [2,2])) 
+0

ありがとうございます。 行数が動的な場合、tf.slice(varTensor、[0、0]、[-1,2])を適用できますか? –

2

をテストすることができtf.slice()

tf.slice(varTensor, [0, 0], [2, 2]); 

を使用することができますPythonの標準スライス表記法に精通していますテンソルフローの使い方を知っている。

関連する問題