张量的拼接

简介: 【8月更文挑战第18天】张量的拼接。

张量的拼接
TensorFlow中,张量拼接的操作主要包括:
tf.contact():将向量按指定维连起来,其余维度不变。
tf.stack() :将一组R维张量变为R+1维张量,拼接前后维度变化。

tf.concat(values, axis, name='concat'):
values:输入张量;
axis:指定拼接维度;
name:操作名称。

代码:
concat_sample_1 = tf.random.normal([4,100,100,3])
concat_sample_2 = tf.random.normal([40,100,100,3])
print("原始数据的尺寸分别为:",concat_sample_1.shape,concat_sample_2.shape)
concated_sample_1 = tf.concat([concat_sample_1,concat_sample_2],axis=0)
print("拼接后数据的尺寸:",concated_sample_1.shape)

输出:
原始数据的尺寸分别为: (4, 100, 100, 3) (40, 100, 100, 3)
拼接后数据的尺寸: (44, 100, 100, 3)

在原来矩阵基础上增加了一个维度,也是同样的道理,axis决定维度增加的位置。
tf.stack(values, axis=0, name='stack'):
values:输入张量;一组相同形状和数据类型的张量。
axis:指定拼接维度;
name:操作名称。

代码:
stack_sample_1 = tf.random.normal([100,100,3])
stack_sample_2 = tf.random.normal([100,100,3])
print("原始数据的尺寸分别为:",stack_sample_1.shape, stack_sample_2.shape)

拼接后维度增加。axis=0,则在第一个维度前增加维度。

stacked_sample_1 = tf.stack([stack_sample_1, stack_sample_2],axis=0)
print("拼接后数据的尺寸:",stacked_sample_1.shape)

输出:
原始数据的尺寸分别为: (100, 100, 3) (100, 100, 3)
拼接后数据的尺寸: (2, 100, 100, 3)

相关文章
|
2月前
矩阵转换
【10月更文挑战第30天】矩阵转换。
32 3
|
6月前
|
机器学习/深度学习 PyTorch 算法框架/工具
|
3月前
|
机器学习/深度学习 PyTorch 算法框架/工具
探索PyTorch:张量的类型转换,拼接操作,索引操作,形状操作
探索PyTorch:张量的类型转换,拼接操作,索引操作,形状操作
|
3月前
|
PyTorch 算法框架/工具 Python
Pytorch学习笔记(十):Torch对张量的计算、Numpy对数组的计算、它们之间的转换
这篇文章是关于PyTorch张量和Numpy数组的计算方法及其相互转换的详细学习笔记。
50 0
|
5月前
|
TensorFlow 算法框架/工具
张量的拼接
【8月更文挑战第17天】张量的拼接。
18 1
|
5月前
|
TensorFlow 算法框架/工具 Python
张量的分割
【8月更文挑战第18天】张量的分割。
35 4
|
5月前
|
TensorFlow 算法框架/工具 索引
张量排序
【8月更文挑战第17天】张量排序。
31 1
|
7月前
|
算法 PyTorch 算法框架/工具
Pytorch - 张量转换拼接
使用 Tensor.numpy 函数可以将张量转换为 ndarray 数组,但是共享内存,可以使用 copy 函数避免共享。
|
机器学习/深度学习 PyTorch TensorFlow
定义张量
张量是一种多维数组,可以用来表示空间中的数据。它通常用于机器学习和人工智能领域,特别是在深度学习中。张量可以看作是多维数组,其中每个元素也是一个数组,可以包含多个元素。
134 1
|
并行计算 安全 异构计算
pytroch取张量数据
pytroch取张量数据
191 1
pytroch取张量数据