Boolean value of Tensor with more than one value is ambiguous

[ljhOfGithub]

the following is my forward:forward(self,pos_fea,subtype,channels),where channels is a parameter of torch.Tensor,and its shape is torch.Size([1]).In the forward,there is a sentence to judge the value of channels:if channels == 2048.
When I use summary(model, [(512,1024,2048),(1,2048),[2048]]) I got the above error,how should I use the summary function to get my expected results?