2020年12月11日
pytorch repeat 和 expand 函数的使用场景，区别
摘要： x = torch.tensor([0, 1, 2, 3]).float().view(4, 1)def test_assign(x): # 赋值操作 x_expand = x.expand(-1, 3) x_repeat = x.repeat(1, 3) x_expand[:, 1] = torc 阅读全文
posted @ 2020-12-11 11:27 佰大于 阅读(681) 评论(0) 推荐(0) 编辑