需转一次np数组,tensor不能与标量相乘。
import numpy as np
import torch
a = torch.ones_like((6,20,1))
print(a.size)
a = a.cpu().numpy()
b = 0.3 * a
b = torch.from_numpy(b)
print(a)
需转一次np数组,tensor不能与标量相乘。
import numpy as np
import torch
a = torch.ones_like((6,20,1))
print(a.size)
a = a.cpu().numpy()
b = 0.3 * a
b = torch.from_numpy(b)
print(a)