Ejemplo n.º 1
0
    class Net(nn.Cell):
        def __init__(self):
            super(Net, self).__init__()
            self.t1 = Tensor([1, 2, 3])

        def construct(self, x):
            return self.t1.expand_as(x)