pytorch中的transpose用法
2023-12-14 12:46:17
注意:维数从0开始,0维 1维2维…,负数代表从右往左数,-1代表第一维,以此类推
import torch
import numpy as np
# 创建一个二维数组
arr = torch.tensor([[[1, 2],
[3, 4]],
[[5, 6],
[7, 8]]])
print("原始数组:")
print(arr)
# torch.transpose转置数组, dim的顺序谁先谁后没有关系,不影响最后结果!!!
# transpose 调换;移项;颠倒顺序
transposed_arr = arr.transpose(0, 1)
print("转置后的数组:")
print(transposed_arr)
print("-----------------------------------------------------")
transposed_arr = arr.transpose(-1, -2)
print("转置后的数组:")
print(transposed_arr)
文章来源:https://blog.csdn.net/zan1763921822/article/details/134842636
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!