如何解决如何填充不同形状的张量以使其在第二维上具有相同的形状
我有3个张量分别为(100,43,1024),(100,37,1024)和(100,42,1024)。我想将所有这些张量的第二维设为第二维的最大值,在这种情况下为43。您能帮助我如何使用垫功能使它们具有相同的形状吗?
解决方法
如果您使用numpy数组,则可以通过以下方式对它们进行零填充:
# create your data
n_sample = 5
X = [np.random.uniform(0,1,(n_sample,43,1024)),np.random.uniform(0,37,42,1024))]
# find max dim
max_dim = np.max([x.shape[1] for x in X])
print(max_dim)
X_pad = []
for x in X:
X_pad.append(np.pad(x,((0,0),(max_dim-x.shape[1],(0,0)),mode='constant')) # pre padding
# X_pad.append(np.pad(x,max_dim-x.shape[1],mode='constant')) # post padding
# check padded shape
print([x.shape for x in X_pad])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。