tf.contrib.framework.assert_same_float_dtype()

tf.contrib.framework.assert_same_float_dtype(tensors=None, dtype=None)

Validate and return float type based on tensors and dtype.

For ops such as matrix multiplication, inputs and weights must be of the same float type. This function validates that all tensors are the same type, validates that type is dtype (if supplied), and returns the type. Type must be dtypes.float32 or dtypes.float64. If neither tensors nor dtype is supplied, default to dtypes.float32.

Args:
  • tensors: Tensors of input values. Can include None elements, which will be ignored.
  • dtype: Expected type.
Returns:

Validated type.

Raises:
  • ValueError: if neither tensors nor dtype is supplied, or result is not float.
doc_TensorFlow
2016-10-14 13:04:44
Comments
Leave a Comment

Please login to continue.