tensorflow 中的 `*` 在哪里记录? [英] Where is `*` documented in tensorflow?

问题描述

我没有找到记录 * 的地方.似乎它可以等同于 tf.multiply 或 tf.scalar_mul.是吗?

I don't find where * is documented. It seems that it can be equivalent to either tf.multiply or tf.scalar_mul. Is it so?

推荐答案

最可靠的文档是 源代码:

def _mul_dispatch(x, y, name=None):
  """Dispatches cwise mul for "Dense*Dense" and "Dense*Sparse"."""
  is_tensor_y = isinstance(y, ops.Tensor)
  if is_tensor_y:
    return gen_math_ops._mul(x, y, name=name)
  else:
    assert isinstance(y, sparse_tensor.SparseTensor)  # Case: Dense * Sparse.
    new_vals = gen_sparse_ops.sparse_dense_cwise_mul(y.indices, y.values,
                                                     y.dense_shape, x, name)
    return sparse_tensor.SparseTensor(y.indices, new_vals, y.dense_shape)

...

_OverrideBinaryOperatorHelper(_mul_dispatch, "mul")

这意味着 __mul__ 运算符重载，它执行 _mul_dispatch.如您所见，它调用了 gen_math_ops._mul(这是 tf.multiply) 或 sparse_dense_cwise_mul 如果张量是稀疏的.

This means __mul__ operator overload, which does _mul_dispatch. As you can see, it calls either gen_math_ops._mul (which is the under-the-hood core function of tf.multiply) or sparse_dense_cwise_mul if the tensor is sparse.

顺便说一下，tf.scalar_mul 只是对 scalar * x 的封装(源代码)，所以基本上是一样的，但依赖是相反的.

By the way, tf.scalar_mul is just a wrapper over scalar * x (source code), so it's basically the same thing, but the dependency is the other way around.

这篇关于tensorflow 中的 `*` 在哪里记录?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！