gpt4 book ai didi

python - 如何向pytorch张量添加新的dim值?

转载 作者:行者123 更新时间:2023-12-04 16:25:24 26 4
gpt4 key购买 nike

在 numpy 我会做

a = np.zeros((4, 5, 6))
a = a[:, :, np.newaxis, :]
assert a.shape == (4, 5, 1, 6)
如何在 torch 中做同样的事情?

最佳答案

a = torch.zeros(4, 5, 6)
a = a[:, :, None, :]

关于python - 如何向pytorch张量添加新的dim值?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/65470807/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com