python - PyTorch 中 "ragged/jagged tensors"的解决方法是什么？

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 的东西这似乎与 RaggedTensors 的目的几乎相同在 tensorflow 中。您可以关注 RFC 并取得进展 here .