hernanbari / pylissom Goto Github PK

from pylissom.nn.modules.lissom import *  
in_features = 25**2 
out_features = 5**2 
cortex = Cortex(in_features, out_features, radius=5, sigma=5)  

../aten/src/ATen/native/LegacyDefinitions.cpp:19: UserWarning: masked_fill_ received a mask with dtype torch.uint8, this behavior is now deprecated,please use a mask with dtype torch.bool instead.
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-2-87d510715ac0> in <module>
      1 in_features = 25**2
      2 out_features = 5**2
----> 3 cortex = Cortex(in_features, out_features, radius=5, sigma=5)

~/Library/Python/3.7/lib/python/site-packages/pylissom/nn/modules/lissom.py in __init__(self, in_features, out_features, radius, sigma)
     29         self.weight.data = normalize(
     30             apply_circular_mask_to_weights(self.weight.data,
---> 31                                            radius=radius))
     32 
     33     def __repr__(self):

~/Library/Python/3.7/lib/python/site-packages/pylissom/math.py in normalize(matrix, norm, axis)
     76     """
     77     matrix_norms = matrix.norm(p=norm, dim=axis)
---> 78     return matrix.div(matrix_norms.expand_as(matrix))

RuntimeError: The expanded size of the tensor (625) must match the existing size (25) at non-singleton dimension 1.  Target sizes: [25, 625].  Tensor sizes: [25]

  tensor = circular_mask(matrix.size()[1], matrix.size()[0], radius, matrix.is_cuda)
  matrix.masked_fill_(tensor, 0)

matrix.masked_fill_(tensor.type(torch.BoolTensor), 0)