如何在 TensorFlow 中删除一行包含 nan 的张量张量
How to drop a row of a tensor tensor containing nan in TensorFlow
我有一个张量：
tensor = tf.convert_to_tensor(np.array([[0, 1, 2, 3, 4, np.nan],
[6, 7, 8, 9, 10, 11]]))
并想删除包含 nan 的行。例如：
<tf.Tensor: shape=(6,), dtype=int64, numpy=array([ 6, 7, 8, 9, 10, 11])>
你是如何在 TensorFlow 中做到这一点的？
此代码将删除 nan 行：
tensor[~np.isnan(tensor).any(axis=1)]
我有一个张量：
tensor = tf.convert_to_tensor(np.array([[0, 1, 2, 3, 4, np.nan],
[6, 7, 8, 9, 10, 11]]))
并想删除包含 nan 的行。例如：
<tf.Tensor: shape=(6,), dtype=int64, numpy=array([ 6, 7, 8, 9, 10, 11])>
你是如何在 TensorFlow 中做到这一点的？
此代码将删除 nan 行：
tensor[~np.isnan(tensor).any(axis=1)]