def _empty(creator: TensorCreator):
return creator.ones((3, 2, 4), dtype=creator.int64)
def _cat_dim_1(creator: TensorCreator):
return creator.cat([creator.zeros((3, 1)), creator.ones((3, 3))], dim=1)
def _ones_single_dim(creator: TensorCreator):
return creator.ones(6, device='cpu')