Tensorflow - あらかじめ定義された値の変形または分割テンソル

Question

2Dテンソルフローテンソルを特定の値（ここではX）に基づいて3Dテンソルフローテンソルを分割し、2番目と3番目にテンソルフローテンソルを埋め込む寸法。例えば

は、我々は以下のテンソルを持っていると言う：

[[1, 2, 3, X, 4, 5, 6], 
[1, 2, X, 5, 6, X, 8], 
[1, 2, 5, X, 8, 9, P]] 

• X（それが0または何か他のものに等しいことができる）に分割すること。
• Pはパディングの記号です。

目的は以下のことです：

[[[1, 2, 3], [4, 5, 6], [P, P, P]], 
[[[1, 2, P], [5, 6, P], [8, P, P]], 
[[[1, 2, 5], [8, 9, P], [P, P, P]] 

Tensorflowはsplit機能を提供しますが、それは、この問題を解決していません。 reshape関数は、分割が特定の値に基づく必要があるため、解決策ではありません。

ありがとうございます。

Answer 1

おそらくtf.py_funcを使用して分割操作を行うことができます。