【python_学习笔记】

TypeError: ‘numpy._DTypeMeta‘ object is not subscriptable

import torch

# 创建一个形状为 (3, 4, 5) 的张量
tensor = torch.randn(3, 4, 5)

# 使用 permute 重新排列维度，例如 (3, 5, 4)
permuted_tensor = tensor.permute(0, 2, 1)

print("Original Tensor Shape:", tensor.shape)
print("Permuted Tensor Shape:", permuted_tensor.shape)

# 在第二个位置添加一个维度，大小为 1 (3, 1, 5, 4)
unsqueezed_tensor = permuted_tensor.unsqueeze(1)

print("Original Permuted Tensor Shape:", permuted_tensor.shape)

print("Unsqueezed Tensor Shape:", unsqueezed_tensor.shape)