Tensor支持从numpy转换为Tensor,并且本身也包含一些与numpy相同的随机函数,比如normal、truncated_normal、uniform、shuffle等函数,需要注意的是gather函数是根据下标来取tensor中指定位置的值,是必用的高频函数.
# coding: utf-8
# @时间 : 2022/3/20 2:59 下午
# @作者 : 那美那美
import tensorflow as tf
import numpy as np
# numpy转换为tensor类型
a1 = tf.convert_to_tensor(np.ones((2, 3))) # np.ones(shape)产生指定维度的0
a2 = tf.ones((2, 3)) # 直接使用tf中的函数
a3 = tf.ones_like(a2) # 根据某个tesnor创建相同维度的0
a4 = tf.fill((2, 2), 1) # 指定维度并填充指定的值1
a5 = tf.random.normal((2, 2), mean=1, stddev=1) # 产生均值为1,标准差为1满足正态分布的随机数
# 生成的值服从具有指定平均值和标准偏差的正态分布,如果生成的值大于平均值2个标准偏差的值则丢弃重新选择
a6 = tf.random.truncated_normal((2, 2), mean=0, stddev=1)
# 生成的值在 [