Python tf.RaggedTensor.numpy用法及代码示例

用法

numpy()

返回一个带有此 RaggedTensor 值的 numpy array 。

要求此 RaggedTensor 是在即刻执行模式下构造的。

参差不齐的维度使用 numpy arrays 和 dtype=object 和 rank=1 进行编码，其中每个元素都是一行。

例子

在以下示例中，RaggedTensor.numpy() 返回的值包含三个 numpy array 对象：每行一个(带有 rank=1 和 dtype=int64 )，一个用于组合它们(带有 rank=1 和 dtype=object ) ：

tf.ragged.constant([[1, 2, 3], [4, 5]], dtype=tf.int64).numpy()
array([array([1, 2, 3]), array([4, 5])], dtype=object)

统一维度使用多维 numpy array s 编码。在以下示例中，RaggedTensor.numpy() 返回的值包含单个 numpy array 对象，带有 rank=2 和 dtype=int64 ：

tf.ragged.constant([[1, 2, 3], [4, 5, 6]], dtype=tf.int64).numpy()
array([[1, 2, 3], [4, 5, 6]])