Amine08:

a=torch.rand(8,8)
a[0:2,:]=2*a[0:2,:]
a[4:6,:]=2*a[4:6,:]



Your solution works fine.

In [2]: a=torch.ones(8,8)
   ...: a[0:2,:]=2*a[0:2,:]
   ...: a[4:6,:]=2*a[4:6,:]

In [3]: a
Out[3]: 
tensor([[2., 2., 2., 2., 2., 2., 2., 2.],
        [2., 2., 2., 2., 2., 2., 2., 2.],
        [1., 1., 1., 1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1., 1., 1., 1.],
        [2., 2., 2., 2., 2., 2., 2., 2.],
        [2., 2., 2., 2., 2., 2., 2., 2.],
        [1., 1., 1., 1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1., 1., 1., 1.]])

In [4]: 

Using a simple timing function (e.g. importing time ) you can measure the speed both ways.