Definition at line 83 of file splat.py.
◆ __init__()
| fastreid.layers.splat.rSoftMax.__init__ |
( |
| self, |
|
|
| radix, |
|
|
| cardinality ) |
Definition at line 84 of file splat.py.
84 def __init__(self, radix, cardinality):
85 super().__init__()
86 self.radix = radix
87 self.cardinality = cardinality
88
◆ forward()
| fastreid.layers.splat.rSoftMax.forward |
( |
| self, |
|
|
| x ) |
Definition at line 89 of file splat.py.
89 def forward(self, x):
90 batch = x.size(0)
91 if self.radix > 1:
92 x = x.view(batch, self.cardinality, self.radix, -1).transpose(1, 2)
93 x = F.softmax(x, dim=1)
94 x = x.reshape(batch, -1)
95 else:
96 x = torch.sigmoid(x)
97 return x
◆ cardinality
| fastreid.layers.splat.rSoftMax.cardinality |
◆ radix
| fastreid.layers.splat.rSoftMax.radix |
The documentation for this class was generated from the following file:
