bc601d05003fe26dfe1a14bb8a3ec61757444ae1,batchflow/models/eager_torch/layers/resize.py,ReduceDim,forward,#ReduceDim#Any#,28

Before Change

        self.dim = dim

    def forward(self, x):
        dim = max(len(get_shape(x)) -  - self.dim, 0)
        ones = [1] * dim
        return x.view(x.size(0), -1, *ones)

After Change

        self.dim = dim

    def forward(self, x):
        dim = max(get_num_dims(x) - self.dim, 0)
        ones = [1] * dim
        return x.view(x.size(0), -1, *ones)

Instances