core_test.py 文件源码

def setUp(self):
    tensor = tf.ones([7, 3, 8, 1])
    a0 = ('x', range(7))
    a1 = ('channel', ['red', 'green', 'blue'])
    a2 = ('y', 8)
    a3 = ('z', tf.Dimension(1))

    self.lt = core.LabeledTensor(tensor, [a0, a1, a2, a3])