Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- inputs = torch.rand(3,12,12)
- conv = nn.SpatialConvolution(3,4,5,5)
- output = conv:forward(inputs)
- print(unpack(output:size():totable()))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement