tf.ragged.map_flat_values
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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))
以上是关于tf.ragged.map_flat_values的主要内容,如果未能解决你的问题,请参考以下文章