Exemplo n.º 1
0
    def setUp(self):
        super(ShuffleBatchTest, self).setUp()

        tensors = []
        for i in range(10):
            offset_lt = core.LabeledTensor(constant_op.constant(i), [])
            tensors.append(core.add(self.original_lt, offset_lt))
        self.pack_lt = ops.pack(tensors, 'batch')
Exemplo n.º 2
0
  def setUp(self):
    super(ShuffleBatchTest, self).setUp()

    tensors = []
    for i in range(10):
      offset_lt = core.LabeledTensor(constant_op.constant(i), [])
      tensors.append(core.add(self.original_lt, offset_lt))
    self.pack_lt = ops.pack(tensors, 'batch')