python - PyTorch 中“参差不齐/锯齿状张量”的解决方法是什么？

Question

Tensorflow 提供了参差不齐的张量（https://www.tensorflow.org/guide/ragged_tensor）。然而，PyTorch 不提供这样的数据结构。是否有一种解决方法可以在 PyTorch 中构建类似的东西？

import numpy as np
x = np.array([[0], [0, 1]])
print(x)  # [list([0]) list([0, 1])]

import tensorflow as tf
x = tf.ragged.constant([[0], [0, 1]])
print(x)  # <tf.RaggedTensor [[0], [0, 1]]>

import torch
# x = torch.Tensor([[0], [0, 1]])  # ValueError

Answer 1

PyTorch 正在实现一个名为的东西NestedTensors，它似乎与RaggedTensorsTensorflow 中的目的几乎相同。您可以在此处关注 RFC 和进度。