2022年7月8日
torch.roll 解读
摘要： torch.roll顾名思义，就是让张量滚动的意思。 形参形式： 1 torch.roll(input, shifts, dims=None) → Tensor 1. input为输入张量. 2. shifts表示要滚动的方向。负数表示左上方向，正数表示右下方向。dims表示要滚动的维度。 如果sh 阅读全文
posted @ 2022-07-08 16:09 重大的小鸿 阅读(949) 评论(0) 推荐(0) 编辑