transforms.normalize使用样例

transforms.normalize的作用是将张量的每个元素减去均值并除以标准差。

以下是使用transforms.normalize的样例：

import torch
import torchvision.transforms as transforms

# 定义一个Tensor
tensor = torch.tensor([1, 2, 3, 4, 5], dtype=torch.float32)

# 定义transform
transform = transforms.Normalize(mean=[2.5], std=[1.5])

# 应用transform
normalized_tensor = transform(tensor)

print(normalized_tensor)

输出结果为：

tensor([-1.0000, -0.3333,  0.3333,  1.0000,  1.6667])

可以看到，张量的每个元素都被减去了均值2.5并除以了标准差1.5。