code: def z2_prob_distribution(size):
      """Samples a binary tensor with uniform probability of 0 and 1.
  
      Args:
          size (int): Number of values to sample.
      """
      return (torch.rand(size) > 0.5).int()
  
imports: from typing import Callable
  import torch
label_desc: Use image node_id 3 for calling a global function with example usage: z2_prob_distribution(size) and returns: unknown