tensor = torch.ones(4, 4)
print('First row: ',tensor[0])
print('First column: ', tensor[:, 0])
print('Last column:', tensor[..., -1])
tensor[:,1] = 0
print(tensor)

pytorch中的矩阵乘和点乘_其他

 

下面是pytorch中的矩阵乘，三种方式结果相同



y1 = tensor @ tensor.T
y2 = tensor.matmul(tensor.T)

y3 = torch.rand_like(tensor)
torch.matmul(tensor, tensor.T, out=y3)

输出

pytorch中的矩阵乘和点乘_pytorch_02

下面是pytorch中的点乘，三种方式结果相同


z1 = tensor * tensor
z2 = tensor.mul(tensor)

z3 = torch.rand_like(tensor)
torch.mul(tensor, tensor, out=z3)

输出

pytorch中的矩阵乘和点乘_人工智能_03