张量变换

2017-06-01  本文已影响0人  虚世界造物主

注意:使用Tensor参数的函数也可以接受tf.convert_to_tensor接受的任何内容。

接口列表整理

类型转换

    tf.string_to_number(string_tensor, out_type=None, name=None)

    说明:将输入Tensor中的每个字符串转换为指定的数字类型,输出一个新Tensor    

    tf.to_double(x, name='ToDouble')

    说明:根据输入的Tensor新建一个相同的Tensor,类型为float64

    tf.to_float(x, name='ToFloat')

说明:根据输入的Tensor新建一个相同的Tensor,类型为float32

    tf.to_bfloat16(x, name='ToBFloat16')

说明:根据输入的Tensor新建一个相同的Tensor,类型为bfloat16

    tf.to_int32(x, name='ToInt32')

说明:根据输入的Tensor新建一个相同的Tensor,类型为int32

    tf.to_int64(x, name='ToInt64')

说明:根据输入的Tensor新建一个相同的Tensor,类型为int64

    tf.cast(x, dtype, name=None)

说明:根据输入的Tensor新建一个相同的Tensor,类型为参数指定类型

维度相关

    tf.shape(input, name=None)

说明:返回输入Tensor的维度信息,返回值本身是1维的Int32类型Tensor

    tf.size(input, name=None)

说明:返回输入Tensor的内部元素个数,返回值是1个Int32类型的数字

    tf.rank(input, name=None)

说明:返回输入Tensor的等级,也就是定位到某个内部元素所需要的索引值个数

    tf.reshape(tensor, shape, name=None)

说明:用输入Tensor内部的元素,按指定维度要求生成一个新Tensor

    tf.squeeze(input, squeeze_dims=None, name=None)

说明:删除输入Tensor的所有1维,后面的参数是如果不想全删,指定需要删除的1维的位置,输出一个新的Tensor

    tf.expand_dims(input, dim, name=None)

说明:在输入Tensor维度的指定位置插入一个尺寸为1的维度,输出一个新的Tensor

裁剪和合并

    tf.slice(input_, begin, size, name=None)

说明:从输入Tensor中,begin位置开始取出size大小的内容,输出一个新Tensor

    tf.split(split_dim, num_split, value, name='split')

说明:从输入Tensor中,指定split_dim维度分割成num_split份数,并输出num_split个新Tensor

    tf.tile(input, multiples, name=None)

说明:将输入的Tensor,按指定维度指定倍数进行复制扩展,输出新Tensor

    tf.pad(input, paddings, name=None)

说明:将输入的Tensor,在每一个维度上根据paddings设定填充0,输出新Tensor

    tf.concat(concat_dim, values, name='concat')

说明:将输入的Tensor,按指定维度进行连接,输出新Tensor

    tf.pack(values, name='pack')

说明:将输入的Tensor数组,合并成一个Tensor,以原数组索引作为合并后Tensor的新维度

    tf.unpack(value, num=None, name='unpack')

说明:tf.pack的逆操作

    tf.reverse_sequence(input, seq_lengths, seq_dim, name=None)

说明:将输入的Tensor按seq_lengths指定的长度左右翻转,输出新Tensor

    tf.reverse(tensor, dims, name=None)

说明:将输入的Tensor按dims指定的维度左右翻转,输出新Tensor

    tf.transpose(a, perm=None, name='transpose')

说明:将输入的Tensor按perm指定的维度顺序进行转置

    tf.gather(params, indices, name=None)

说明:从输入的Tensor中按indices指定的位置取出值,拼接成一个新的Tensor

tf.dynamic_partition(data, partitions, num_partitions, name=None)

说明:根据partitons的内容是0还是1,对输入的data的内容进行分类。依次取出对应的内容组合成新的多个Tensor。

tf.dynamic_stitch(indices, data, name=None)

说明:根据indices列表指定的位置重新排列data内的元素,输出新Tensor

上一篇下一篇

猜你喜欢

热点阅读