tf.ragged.map_flat_values

tf.ragged.map_flat_values 可以理解不规则矩阵的map

import tensorflow as tf 


rt = tf.ragged.constant([[1, 2, 3], [], [4, 5], [6]])

tf.ragged.map_flat_values(tf.ones_like, rt).to_list()
tf.ragged.map_flat_values(tf.multiply, rt, rt).to_list()

tf.ragged.map_flat_values 比下面更通用

import numpy as np

import tensorflow as tf
elems = np.array([1, 2, 3, 4, 5, 6])
squares = tf.map_fn(lambda x: x * x, elems)

print(squares)

 ids_map_layer = tf.keras.layers.Lambda(lambda x:tf.ragged.map_flat_values(ids_table.lookup, x))