如何根据Tensorflow中的条件提出错误？

Question

我正在使用tf.py_函数编写一个函数my_函数，用作tf.data.Dataset.map参数，如the doc中所建议的那样

我如何在我的函数中提出一个tf.errors.InvalidArgumentError

我的功能

def my_function(data, param):
    _, n_freq_bins, n_time_bins, channels = tf.shape(data)
    tf.cond(
        param < n_time_bins,
        lambda: 1,
        <raise tf.errors.InvalidArgumentError>
    )
    ...

在我的数据管道中：

dataset = dataset.map(lambda data, labels: (tf.py_function(my_function, [data, 100], tf.float32), labels))