class Net(nn.Cell):
def __init__(self):
super(Net, self).__init__()
self.value = Tensor([[1, 2, 3], [4, 5, 6]], dtype=mstype.float32)
def construct(self):
return self.value.flatten(order=123)
class Net(nn.Cell):
def __init__(self):
super(Net, self).__init__()
self.value = Tensor([[1, 2, 3], [4, 5, 6]], dtype=mstype.float32)
def construct(self):
return self.value.flatten(order=123)
