标签： tensorflow tensorflow-datasets
我有一个数据集，我正在使用负抽样训练模型。每个示例都有一些ID，对应于将要使用的一些固定的负样本。
除了固定的阴性样本，我想包括随机阴性样本。但是，每个示例都有可变数量的固定负样本，因此我需要附加可变数量的随机ID。这基本上是由padded_batch函数执行的函数，但我需要填充来自某个范围的随机整数。
padded_batch
有没有办法通过Tensorflow Python API实现这一目标，还是需要用C ++编写自定义数据集类？