data exist,
but plot is empty
import matplotlib.pyplot as plt

train_y = torch.tensor([[43], [0]]).cuda()
num = train_y[0]
print(num)

data = torch.randn(50,3)
print(data[num])

plt.figure(0)
plt.plot(data[num])

print(data[43])

plt.figure(1)
plt.plot(data[43])

data[num][0]
THANKS