tf.ones

tf.ones 함수는 모든 요소가 1인 텐서를 생성합니다.

예제

import tensorflow as tf

a = tf.ones(3)
b = tf.ones([4])
c = tf.ones([2, 2, 2])

print(a)
print(b)
print(c)

tf.Tensor([1. 1. 1.], shape=(3,), dtype=float32)
tf.Tensor([1. 1. 1. 1.], shape=(4,), dtype=float32)
tf.Tensor(
[[[1. 1.]
[1. 1.]]

[[1. 1.]
[1. 1.]]], shape=(2, 2, 2), dtype=float32)

tf.ones()는 모든 요소가 1인 텐서를 만듭니다.

tf.ones()에 만들어질 텐서의 형태 (shape)를 입력합니다.

이전글/다음글

이전글 : tf.ones_initializer

다음글 : tf.random_normal_initializer

공유하기