matmul_with_broadcast added to linear_operator_util. This is like matmul, but
will work for non-matching batch dimensions. This is achieved by broadcasting, by adding a Tensor of zeros to arguments that cannot be determined to already be of the broadcast shape. Change: 151631491
Loading
Please sign in to comment