class_docstr: Arbitrary tangent space when no more-efficient special case applies.
class_docstr_tok: ["Arbitrary","tangent","space","when","no","more-efficient","special","case","applies","."]
code: def _transform_coordinatewise(self, x, f, event_ndims=None, **kwargs):
      diag_jacobian = coordinatewise_jvp(f, x)
      diag_linop = tf.linalg.LinearOperatorDiag(diag_jacobian)
      if event_ndims is None:
          event_ndims = f.forward_min_event_ndims
      if isinstance(self.basis, LinearOperatorBasis):
          new_basis = LinearOperatorBasis(
              self.basis.basis_linop @ diag_linop
          )
      else:
          diag_jacobian_reshape = _reshape_to_matrix(
              diag_jacobian[tf.newaxis, ...], event_ndims
          )
          new_basis = LinearOperatorBasis(
              tf.linalg.LinearOperatorFullMatrix(
                  self.basis.to_dense_matrix(event_ndims)
                  * diag_jacobian_reshape
              )
          )
      return self._transform_from_basis(new_basis, event_ndims)
  
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.internal import distribution_util
  from tensorflow_probability.python.internal import nest_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import tensor_util
  from tensorflow_probability.python.internal import tensorshape_util
label_desc: Use image node_id 4 for calling the GeneralSpace obj's underlying member method code with example usage: obj._transform_coordinatewise(x, f, event_ndims) and returns: self