Random Tensor Generation Utility

  • Share this:

Code introduction


This function generates a random tensor of the specified shape and data type, usually used for testing or data generation.


Technology Stack : TensorFlow, NumPy

Code Type : The type of code

Code Difficulty :


                
                    
import tensorflow as tf
import numpy as np

def generate_random_tensor(shape, dtype=tf.float32):
    """
    Generate a random tensor of the specified shape and data type.
    """
    return tf.random.uniform(shape, dtype=dtype)