比方說,我在下面使用caffe.Netspec()
設計網絡(例如,在盜夢空間)。當設計網絡`caffe.NetSpec()',怎麼知道某個斑點形狀?
import caffe
from caffe import layers as L
net = caffe.NetSpec()
net.data = ...
net.label = ...
net["conv1"] = L.convolution(...)
# ... concatenations ... and so on.
net["here"] = ...
有沒有辦法知道任何特定層(例如,層here
)等7x7x128
的團塊形狀? 我想打它,直到1x1
形狀,所以我需要知道哪些形狀爲止。