div operation occur error

[zlbing]

I have a tensor A shape [1,256,15,20] and B shape [1,1,15,20],
when I use A.div(B), it error.
the error is
terminate called after throwing an instance of 'std::runtime_error' what(): invalid argument 3: sizes do not match at /home/zzz/work-space/pytorch-cpp/ATen/lib/THC/generated/../generic/THCTensorMathPointwise.cu:345

for Example
Tensor A = CPU(kFloat).randn({1,10,5,5}); Tensor B = CPU(kFloat).randn({1,1,5,5}); A = A.div(B);