Python tensor示例

示例#1

def test_resetnet50_build():
    X = me.tensor()
    Y = me.tensor()
    X.set_dtype(dtype.float32)
    Y.set_dtype(dtype.float32)
    network = resnet50()
    model = Model(network=network, loss_fn=loss_func, optimizer=optimizer)

示例#2

def test_tensor_add():
    X = me.tensor()
    Y = me.tensor()
    X.set_dtype(dtype.float32)
    Y.set_dtype(dtype.float32)
    X = me.tensor(np.ones([2, 3]))
    Y = me.tensor(np.ones([2, 3]))
    sum = add(X, Y)
    print("test tensor add")
    return sum