FloatTensor # 定义默认GPU 的数据类型 gpu_tensor = torch.randn(10, ... x = torch.randn(3, 4, 5) print(x.shape) # 使用permute和transpose进行维度交换 x ...
確定! 回上一頁