我有一個 4-D 張量形狀[6,20,30,6]，我想執行 keras/tensorflow 等效于：new = np.array([old[i,:,:,i] for i in range(6)])任何幫助表示贊賞！
2 回答
慕蓋茨4494581
TA貢獻1850條經驗 獲得超11個贊
我能夠在純 keras 中進行轉換。
def cc(x): return K.backend.stack([x[:,i, :, :, i] for i in range(6)], axis=1)
然后在 keras 模型定義中：
new=L.Lambda(lambda y: cc(y))(old)
添加回答
舉報
0/150
提交
取消