mocks.py 文件源码

def __init__(self, name, dtype):
    super(MockSparseTensor, self).__init__()
    self._name = name
    self._dtype = dtype
    self._shape = tensor_shape.unknown_shape()
    self.indices = MockTensor("%s indices" % name, dtypes.int32)
    self.values = MockTensor("%s values" % name, dtype)