2024年8月10日
命名张量的解释
摘要： import torch 把3个颜色grb通道合并为一个灰度通道 定义变量，用随机数模拟 img_t = torch.randn(3, 5, 5) batch_t = torch.randn(2, 3, 5, 5) weights = torch.randn(3) 朴素的求法 以 channels 阅读全文
posted @ 2024-08-10 16:32 立体风 阅读(16) 评论(0) 推荐(0) 编辑