摘要：
生成值范围在[0,1]的张量 a = torch.rand(5, 2, 3) 元素总数 a.numel() # 30 获取张量的形状 a.size() a.shape 取值(不参与计算) a.data # 不安全 a.detach() # 推荐 平均值,最大值,最小值 a.detach().mean 阅读全文
摘要：
图片保存工具 # 图片张数 N = 16 # 通道数 C = 3 # 高度 H = 64 # 宽度 W = 32 image = torch.rand(N,C,H,W) # 路径 path = './' save_image(image, path) 阅读全文