我想使用創建 Tensorflow 資料集,tf.data.Dataset.from_tensor_slices但出現此錯誤:
Can't convert non-rectangular Python sequence to Tensor.
為了簡化問題,我采用了與我的資料類似的以下示例:
import tensorflow as tf
data = ['A', 'B']
label = [ ['a1', 'a2', 'a3'] , ['b1', 'b2', 'b3' , 'b4' ] ]
dataset = tf.data.Dataset.from_tensor_slices((data , label))
出現此問題是因為標簽中的串列不相等,len(['a1', 'a2', 'a3']) not equal len(['b1', 'b2', 'b3' , 'b4' ]).
我想保持資料不變而不使用填充。我在這個網站上嘗試tf.ragged.constant了其他解決方案,但它對我不起作用。
uj5u.com熱心網友回復:
這應該作業,
dataset = tf.data.Dataset.from_tensor_slices((data , tf.ragged.constant(label)))
for data in dataset.as_numpy_iterator():
print(data)
#outputs
(b'A', array([b'a1', b'a2', b'a3'], dtype=object))
(b'B', array([b'b1', b'b2', b'b3', b'b4'], dtype=object))
轉載請註明出處,本文鏈接:https://www.uj5u.com/gongcheng/524100.html
標籤:Python张量流数据集
上一篇:InvalidArgumentError:輸入檔案名張量必須是標量,但具有形狀:[1][Op:ReadFile]
