对于所有的层,
b
a
t
c
h
S
i
z
e
≤
16
batchSize \leq 16
batchSize≤16 ,
c
h
a
n
n
e
l
s
≤
4096
channels\leq4096
channels≤4096 ,并且特殊维度H和W也
≤
4096
\leq4096
≤4096。
包含padding在内,网络中的
t
e
n
s
o
r
≤
2
31
−
1
tensor\leq2^{31}-1
tensor≤231−1,请注意,如果通道大小不是矢量大小的精确倍数,则向量化格式具有填充。例如,kCHW32格式如果C=1将占用32倍的空间。