linear scale function Args: * w: A Tensor (weights or activation output), the shape is [bucket_size, bucekt_num] if use_buckets else the original size. * mode: A string, 'weight' or 'activation' Returns: * A Tensor, the normalized weights * A Tensor, alpha, scalar
(self, w, mode)
source not stored for this graph (policy: none)