Функция nn.bias_add
Функция nn.bias_add добавляет вектор
смещения к последней оси входного тензора.
Первым параметром функция принимает тензор,
вторым - вектор смещения. Третьим параметром
можно передать формат данных, а четвертым -
имя операции. Функция часто используется
после сверточного или полносвязного слоя.
Синтаксис
tf.nn.bias_add(value, bias, [data_format], [name])
Пример
Давайте добавим вектор смещения
1, 2, 3 к тензору
1, 2, 3:
import tensorflow as tf
t = tf.constant([1, 2, 3])
bias = tf.constant([1, 2, 3])
res = tf.nn.bias_add(t, bias)
print(res)
Результат выполнения кода:
tf.Tensor([2 4 6], shape=(3,), dtype=int32)
Пример
Давайте добавим вектор смещения к двумерному тензору:
import tensorflow as tf
t = tf.constant([[1, 2, 3], [4, 5, 6]])
bias = tf.constant([1, 1, 1])
res = tf.nn.bias_add(t, bias)
print(res)
Результат выполнения кода:
tf.Tensor(
[[2 3 4]
[5 6 7]], shape=(2, 3), dtype=int32)
Пример
Давайте применим функцию к тензору
с типом float32:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5], dtype=tf.float32)
bias = tf.constant([0.5, 0.5, 0.5, 0.5, 0.5])
res = tf.nn.bias_add(t, bias)
print(res)
Результат выполнения кода:
tf.Tensor([1.5 2.5 3.5 4.5 5.5], shape=(5,), dtype=float32)
