提交 b27c59d1 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Simplify `expand_empty`

上级 c70a8869
...@@ -231,8 +231,8 @@ def expand_empty(tensor_var, size): ...@@ -231,8 +231,8 @@ def expand_empty(tensor_var, size):
if size == 0: if size == 0:
return tensor_var return tensor_var
shapes = [tensor_var.shape[x] for x in range(tensor_var.ndim)] shapes = tuple(tensor_var.shape)
new_shape = [size + shapes[0]] + shapes[1:] new_shape = (size + shapes[0], *shapes[1:])
empty = AllocEmpty(tensor_var.dtype)(*new_shape) empty = AllocEmpty(tensor_var.dtype)(*new_shape)
ret = set_subtensor(empty[: shapes[0]], tensor_var) ret = set_subtensor(empty[: shapes[0]], tensor_var)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论