mxgrahp 获取画布
时间: 2024-02-18 09:31:25 浏览: 90
商业模式画布解读
MXNet Gluon中的mxgraph可以通过创建一个mxgraph.plot.Plot实例来获取画布。例如:
```
import mxnet as mx
from mxboard import SummaryWriter
from mxboard import gluon
from mxnet.gluon import nn
# 创建一个模型
net = nn.Sequential()
net.add(nn.Dense(10, activation='relu'))
net.add(nn.Dense(2))
# 定义输入
input_data = mx.nd.random.uniform(shape=(10, 5))
# 创建一个mxgraph.plot.Plot实例,设置参数
plot = mx.gluon.contrib.plotting.plot_network(net, title='My Network')
# 获取画布
canvas = plot.plot(input_data)
```
在这个例子中,我们首先定义了一个简单的神经网络模型`net`,然后定义了一个输入`input_data`。接着,我们创建了一个`mxgraph.plot.Plot`实例`plot`,并使用`mx.gluon.contrib.plotting.plot_network`函数将模型绘制到画布上。最后,我们通过调用`plot.plot(input_data)`方法获取画布。
阅读全文