What is number input for visulizer model?

[tphankr]

I run this code in Jupyter Notebook，but one error occurs:
`import torch
import torchvision.models
import hidden layer as hl
model = torchvision.models.vgg16()
hl.build_graph(model, torch.zeros([1, 3, 224, 224]))'

And now, I want to feed three inputs are [ 32,6, 25, 25] [ 32,6, 51, 51],[32, 6, 75, 75]. I wrote : hl.build_graph(net, torch.zeros(
[ 32,6, 25, 25],
[ 32,6, 51, 51],
[32, 6, 75, 75]
)

But my code had the error: TypeError: zeros(): argument 'out' (position 2) must be Tensor, not list.
How do I fix this problem? (I also tried many ways, ex:
#hl.build_graph(net, torch.zeros( 32,6, 25, 25), torch.zeros(32, 6, 51, 51), torch.zeros(32, 6, 75, 75))
#hl.build_graph(net, torch.zeros(( 32,6, 25, 25), ( 32,6, 51, 51), ( 32,6, 75, 75)))
#hl.build_graph(net, torch.zeros([( 32,6, 25, 25)],[( 32,6, 51, 51)],[( 32,6, 75, 75)])
)

Thank you.

[waleedka]

Closing this because you already solved it in another thread.