torch.arange(12, dtype=torch.float32).reshape(3, 4)
