随机置矩阵列为0[矩阵乘法pytorch版]

1. 举例：

$$\begin{array}{c}A=\left[\begin{array}{cccccccc}0& 1& 2& 3& 4& 5& 6& 7\\ & & & & & & & \\ 8& 9& 10& 11& 12& 13& 14& 15\\ & & & & & & & \\ 16& 17& 18& 19& 20& 21& 22& 23\\ & & & & & & & \\ 24& 25& 26& 27& 28& 29& 30& 31\\ & & & & & & & \\ 32& 33& 34& 35& 36& 37& 38& 39\\ & & & & & & & \\ 40& 41& 42& 43& 44& 45& 46& 47\end{array}\right],rand=[0,5,2]\to A=\left[\begin{array}{cccccccc}0& 1& 0& 3& 4& 0& 6& 7\\ & & & & & & & \\ 0& 9& 0& 11& 12& 0& 14& 15\\ & & & & & & & \\ 0& 17& 0& 19& 20& 0& 22& 23\\ & & & & & & & \\ 0& 25& 0& 27& 28& 0& 30& 31\\ & & & & & & & \\ 0& 33& 0& 35& 36& 0& 38& 39\\ & & & & & & & \\ 0& 41& 0& 43& 44& 0& 46& 47\end{array}\right]\end{array}$$

2. python 代码

import torch
import torch.nn as nntorch.manual_seed(234)class RandomPermute():def __init__(self, matrix, num):super(RandomPermute, self).__init__()self.matrix = matrixself.row, self.column = self.matrix.shapeself.num = numdef get_result(self):if self.num > self.column:set_num = self.columnelse:set_num = self.nummy_permute = torch.randperm(self.column)[:set_num]my_ones = torch.eye(self.column)print(f"before:my_ones=\n{my_ones}")my_ones[:, my_permute] = 0print(f"after:my_ones=\n{my_ones}")print(f"matrix=\n{self.matrix}")print(f"my_permute={my_permute}")my_result = self.matrix @ my_onesprint(f"my_result=\n{my_result}")return my_resultif __name__ == "__main__":run_code = 0a_matrix = torch.arange(48).reshape(6, 8).to(torch.float)set_num = 3my_rand_perm = RandomPermute(a_matrix, set_num)out_result = my_rand_perm.get_result()

• 结果：

before:my_ones=
tensor([[1., 0., 0., 0., 0., 0., 0., 0.],[0., 1., 0., 0., 0., 0., 0., 0.],[0., 0., 1., 0., 0., 0., 0., 0.],[0., 0., 0., 1., 0., 0., 0., 0.],[0., 0., 0., 0., 1., 0., 0., 0.],[0., 0., 0., 0., 0., 1., 0., 0.],[0., 0., 0., 0., 0., 0., 1., 0.],[0., 0., 0., 0., 0., 0., 0., 1.]])
after:my_ones=
tensor([[0., 0., 0., 0., 0., 0., 0., 0.],[0., 1., 0., 0., 0., 0., 0., 0.],[0., 0., 0., 0., 0., 0., 0., 0.],[0., 0., 0., 1., 0., 0., 0., 0.],[0., 0., 0., 0., 1., 0., 0., 0.],[0., 0., 0., 0., 0., 0., 0., 0.],[0., 0., 0., 0., 0., 0., 1., 0.],[0., 0., 0., 0., 0., 0., 0., 1.]])
matrix=
tensor([[ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.],[ 8.,  9., 10., 11., 12., 13., 14., 15.],[16., 17., 18., 19., 20., 21., 22., 23.],[24., 25., 26., 27., 28., 29., 30., 31.],[32., 33., 34., 35., 36., 37., 38., 39.],[40., 41., 42., 43., 44., 45., 46., 47.]])
my_permute=tensor([0, 5, 2])
my_result=
tensor([[ 0.,  1.,  0.,  3.,  4.,  0.,  6.,  7.],[ 0.,  9.,  0., 11., 12.,  0., 14., 15.],[ 0., 17.,  0., 19., 20.,  0., 22., 23.],[ 0., 25.,  0., 27., 28.,  0., 30., 31.],[ 0., 33.,  0., 35., 36.,  0., 38., 39.],[ 0., 41.,  0., 43., 44.,  0., 46., 47.]])