人工智能AI

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)
# 生成的值在 [minval, maxval) 范围内遵循均匀分布。该范围包括下限 minval ，而上限