提交 8cdf8e62 authored 作者: Tim Cooijmans's avatar Tim Cooijmans

BatchedDot: satisfy flake8's perverse desires

上级 d8538e4a
......@@ -3541,8 +3541,10 @@ class BatchedDot(Op):
# generate code to allocate output based on runtime input shapes
z_dims = ["PyArray_DIMS(%s)[0]" % _x]
if x_ndim == 3: z_dims.append("PyArray_DIMS(%s)[1]" % _x)
if y_ndim == 3: z_dims.append("PyArray_DIMS(%s)[2]" % _y)
if x_ndim == 3:
z_dims.append("PyArray_DIMS(%s)[1]" % _x)
if y_ndim == 3:
z_dims.append("PyArray_DIMS(%s)[2]" % _y)
assert len(z_dims) == z_ndim
z_shape_correct = " && ".join("PyArray_DIMS(%s)[%i] == %s"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论