def one_generator(shape, dtype):
return generators.constant(np.ones(shape, dtype=dtype))
def identity_generator(shape, dtype):
assert len(shape) == 2 and shape[0] == shape[1]
return generators.constant(np.identity(shape[0], dtype=dtype))
def zero_generator(shape, dtype):
return generators.constant(np.zeros(shape, dtype=dtype))