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的主要内容,如果未能解决你的问题,请参考以下文章