2022年9月18日
pytorch如何批量reshape，如何每batch_size进行reshape
摘要： 假设我有一个tensor，它的batch_size是2： tensor = torch.randn([2, 6]) print(tensor.shape) 输出是 torch.Size([2, 6]) 其中tensor.shape[0]代表tensor的batch_size 如果我要把其中每个Bat 阅读全文
posted @ 2022-09-18 10:11 猪猪猪猪侠 阅读(351) 评论(0) 推荐(0) 编辑