Function random_tensor

numpy_ml/utils/testing.py:77–88

Create a random real-valued tensor of shape `shape`. If `standardize` is True, ensure each column has mean 0 and std 1.

(shape, standardize=False)

def random_tensor(shape, standardize=False):
    """
    Create a random real-valued tensor of shape `shape`. If `standardize` is
    True, ensure each column has mean 0 and std 1.
    """
    offset = np.random.randint(-300, 300, shape)
    X = np.random.rand(*shape) + offset

    if standardize:
        eps = np.finfo(float).eps
        X = (X - X.mean(axis=0)) / (X.std(axis=0) + eps)
    return X


def random_binary_tensor(shape, sparsity=0.5):