Pytorch-IndexError: too many indices for tensor of dimension 0

在采用pytorch进行CNN实现了，遇到了如下的问题：
IndexError: too many indices for tensor of dimension 0
具体代码如下：

JitterTest_Dataset = JitterDataset(
    csv_file= './jitter_labels/new_jitter_test.CSV',
    root_dir= './Test_jitter_Square/',
    transform= transforms.Compose([
        transforms.Grayscale(num_output_channels=1),
        transforms.ToTensor(),
        transforms.Normalize((0.5),(0.5,)),
)

具体的问题在于：

transforms.Normalize((0.5),(0.5,)),

需要在第一个0.5后面加上逗号。
问题分析：
可能是pytorch版本导致的。
将代码改为：

transforms.Normalize((0.5,),(0.5,)),

问题解决~