提交 87ee6949 编写于 作者: Q qiaolongfei

complete the demo code

上级 2695d96f
...@@ -50,7 +50,7 @@ message LoDTensorDesc { ...@@ -50,7 +50,7 @@ message LoDTensorDesc {
In Python API, layer will take Variable as Input, and return Variable as Output. In Python API, layer will take Variable as Input, and return Variable as Output.
```python ```python
image = Variable() image = Variable(dims=[-1, 640, 480])
# fc1 and fc2 are both Variable # fc1 and fc2 are both Variable
fc1 = layer.fc(input=image, output_size=10) fc1 = layer.fc(input=image, output_size=10)
fc2 = layer.fc(input=fc1, output_size=20) fc2 = layer.fc(input=fc1, output_size=20)
...@@ -90,14 +90,14 @@ def flatten_size(X, num_flatten_dims): ...@@ -90,14 +90,14 @@ def flatten_size(X, num_flatten_dims):
return prod return prod
def layer.fc(X, output_size, num_flatten_dims): def layer.fc(X, output_size, num_flatten_dims):
W = Var(type=FP32, dims=[flatten_size(X, num_flatten_dims), output_size]) W = Variable(type=FP32, dims=[flatten_size(X, num_flatten_dims), output_size])
b = Variable(type=FP32, dims=[output_size]) b = Variable(type=FP32, dims=[output_size])
out = Variable(type=FP32) out = Variable(type=FP32)
y = operator.fc(X, W, b, output=out) # fc will put fc op input into out y = operator.fc(X, W, b, output=out) # fc will put fc op input into out
pd.InferShape(y) pd.InferShape(y)
return out return out
x = var(dim=[-1, 640, 480]) x = Variable(dims=[-1, 640, 480])
y = layer.fc(x, output_size=100) y = layer.fc(x, output_size=100)
z = layer.fc(y, output_size=200) z = layer.fc(y, output_size=200)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册