B !`^&@sdZddlmZddlmZddlmZddlZddlmZ ddl m Z ddl m Z dd lmZdd lmZd Zdd d ZdddZdddZdS)aPHistogram summaries and TensorFlow operations to create them, V2 versions. A histogram summary stores a list of buckets. Each bucket is encoded as a triple `[left_edge, right_edge, count]`. Thus, a full histogram is encoded as a tensor of dimension `[k, 3]`. In general, the value of `k` (the number of buckets) will be a constant, like 30. There are two edge cases: if there is no data, then there are no buckets (the shape is `[0, 3]`); and if there is data but all points have the same value, then there is one bucket whose left and right endpoints are the same (the shape is `[1, 3]`). )absolute_import)division)print_functionN)tf2) summary_pb2)metadata)lazy_tensor_creator) tensor_utilcsztjd|dttjjddp$tjjfdd}ttj tjjj tjj frltj j j ||||S||||S)aWrite a histogram summary. Arguments: name: A name for this summary. The summary tag used for TensorBoard will be this name prefixed by any active name scopes. data: A `Tensor` of any shape. Must be castable to `float64`. step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. buckets: Optional positive `int`. The output will have this many buckets, except in two edge cases. If there is no data, then there are no buckets. If there is data but all points have the same value, then there is one bucket whose left and right endpoints are the same. description: Optional long-form description for this summary, as a constant `str`. Markdown is supported. Defaults to empty. Returns: True on success, or false if no summary was emitted because no default summary writer was available. Raises: ValueError: if a default writer exists, but no step was provided and `tf.summary.experimental.get_step()` is None. N) display_name description summary_scopec sLd|gd.\}}tjfdd}tjj|||dSQRXdS)Nhistogram_summary)valuescs tS)N)_buckets)bucketsdatarW/home/dcms/DCMS/lib/python3.7/site-packages/tensorboard/plugins/histogram/summary_v2.py lazy_tensorVsz9histogram..histogram_summary..lazy_tensor)tagtensorstepr)rZLazyTensorCreatortfsummarywrite)rrZhistogram_metadatarr_r)namesummary_metadatar )rrrrOsz$histogram..histogram_summary)rcreate_summary_metadatagetattrrrZ experimentalr isinstanceZ distributeZ get_strategyZ TPUStrategycompatZv1ZtpuZoutside_compilation)rrrrr rr)rrr r histogram,s  r#c sdkr ttdttjtjtjtjdgdttj t tj dd}dd}fd d }t |||SQRXdS) aCreate a TensorFlow op to group data into histogram buckets. Arguments: data: A `Tensor` of any shape. Must be castable to `float64`. bucket_count: Optional positive `int` or scalar `int32` `Tensor`. Returns: A `Tensor` of shape `[k, 3]` and type `float64`. The `i`th row is a triple `[left_edge, right_edge, count]` for a single bucket. The value of `k` is either `bucket_count` or `1` or `0`. Nr)shape)inputrcSstjgdtjdS)N)r)r%dtype)rZconstantfloat64rrrr when_emptysz_buckets..when_emptycs\tjdtjdtd}fdd}fdd}t|||S)N) input_tensorrc sttj}}tjt||tjd}t|d}tj|d}tjtj|ddtjd}td}t |ddggd}|dd}|dd}tj t |||gdS)N)r()depthr)r+axisr$)a) rcastr)floorint32minimumZone_hotZ reduce_sumlinspaceconcat transposestack) bucket_widthoffsetsbucket_indicesclamped_indicesone_hots bucket_countsedges left_edges right_edges) bucket_countrmax_min_range_rrwhen_nonsingulars    z9_buckets..when_nonempty..when_nonsingularcsZ}t|dg}t|dg}tttjdtjg}tjt|||gdS)Ng?)r&)r/)rr7r0sizer)r6)centerZ bucket_startsZ bucket_endsr=)rrCrr when_singularsz6_buckets..when_nonempty..when_singular)rZ reduce_minZ reduce_maxequalcond)Z is_singularrErH)rAr)rBrCrDr when_nonemptys    z_buckets..when_nonempty) DEFAULT_BUCKET_COUNTrZ name_scope debuggingZ assert_scalarZ assert_typer2reshaper0r)rIrFrJ)rrAZis_emptyr*rKr)rArrros   )rcCs|dkr tn|}t|t}|jdkrBtgd}n t|}t |}||}|dkr|}t|d|dt|jgg}n||} ||} t | | t } t | |d} t| g td|k} | j|j|fkst| j|j|fftj| dd}t|||d}|dd}|dd}t|||g }tj|tjd}tjd|d }t}|jj|||d |S) aCreate a histogram summary protobuf. Arguments: tag: String tag for the summary. data: A `np.array` or array-like form of any shape. Must have type castable to `float`. buckets: Optional positive `int`. The output will have this many buckets, except in two edge cases. If there is no data, then there are no buckets. If there is data but all points have the same value, then there is one bucket whose left and right endpoints are the same. description: Optional long-form description for this summary, as a `str`. Markdown is supported. Defaults to empty. Returns: A `summary_pb2.Summary` protobuf object. Nr)rr'g?r,)r.r$)r()r r )rrr)rLnparrayflattenZastypefloatrFrNminmaxr1intr3r6Zaranger%AssertionErrorsumr4r Zmake_tensor_protor)rrrZSummaryvalueadd)rrrr rArCrBrDrGr8r9r:r;r<r=r>r?r@rrrrrr histogram_pbs>   "    rZ)NNN)N)NN)__doc__ __future__rrrnumpyrOZtensorboard.compatrrZtensorboard.compat.protorZtensorboard.plugins.histogramrZtensorboard.utilrr rLr#rrZrrrrs         C C