Функция ones
Функция ones создает тензор, заполненный
единицами. Первым параметром функция принимает
форму будущего тензора в виде списка или кортежа.
Вторым параметром можно передать тип данных
элементов, например tf.float32 или tf.int32.
Если тип не указан, по умолчанию используется
tf.float32.
Синтаксис
tf.ones(shape, [dtype])
Пример
Давайте создадим одномерный тензор из пяти единиц:
import tensorflow as tf
t = tf.ones([5])
print(t)
Результат выполнения кода:
tf.Tensor([1. 1. 1. 1. 1.], shape=(5,), dtype=float32)
Пример
Давайте создадим двумерный тензор формы 2 на 3:
import tensorflow as tf
t = tf.ones([2, 3])
print(t)
Результат выполнения кода:
tf.Tensor(
[[1. 1. 1.]
[1. 1. 1.]], shape=(2, 3), dtype=float32)
Пример
Давайте создадим тензор с целочисленным типом int32:
import tensorflow as tf
t = tf.ones([3], dtype=tf.int32)
print(t)
Результат выполнения кода:
tf.Tensor([1 1 1], shape=(3,), dtype=int32)
