pytorch tensor赋值 修改 经典 bug

import torch
import numpy as np

A = torch.tensor([0.1, 0.2, 0.3]).to('cuda')

B= A
B[0] = -0.1
B[1] = -0.5
B[2] = -34

print("B:  ", B)

print("A:  ", A)

输出：

B:   tensor([ -0.1000,  -0.5000, -34.0000], device='cuda:0')
A:   tensor([ -0.1000,  -0.5000, -34.0000], device='cuda:0')