mean = torch.tensor([0.5071, 0.4866, 0.4409]), 无法赋值给函数调用
时间: 2024-03-19 14:45:07 浏览: 42
PyTorch中torch.tensor与torch.Tensor的区别详解
你说的情况可能是由于在函数调用时错误地使用了逗号来分隔变量,导致Python将其解释为一个元组。应该将逗号替换为等号以将mean和std分别赋值为张量对象。例如:
```
mean = torch.tensor([0.5071, 0.4866, 0.4409])
std = torch.tensor([1.8643e-06, 1.2296e-06, 2.5785e-06])
get_mean_std_value(train_loader, mean, std)
```
在这个示例中,我们首先创建了两个张量对象mean和std,并将它们赋值为指定的值。然后我们调用了一个名为get_mean_std_value()的函数,并将train_loader、mean和std作为参数传递给它。这样,函数就可以使用正确的张量对象来计算均值和标准差了。
阅读全文