提交 1f72c472 authored 作者: James Bergstra's avatar James Bergstra

added early exit path to broadcast_like to it can act as identity()

上级 94a52915
...@@ -87,6 +87,9 @@ def broadcast_like(value, template, env, dtype=None): ...@@ -87,6 +87,9 @@ def broadcast_like(value, template, env, dtype=None):
filled by broadcasting value through it. `value` will be casted as necessary. filled by broadcasting value through it. `value` will be casted as necessary.
""" """
value = T.as_tensor_variable(value)
if value.type == template.type:
return value
shape_of = env.shape_feature.shape_of shape_of = env.shape_feature.shape_of
if template not in shape_of: if template not in shape_of:
raise NotImplementedError('broadcast_like currently requires the template Variable to be in the env already') raise NotImplementedError('broadcast_like currently requires the template Variable to be in the env already')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论