Python PyTorch Tensor.rename用法及代碼示例

本文簡要介紹python語言中 torch.Tensor.rename 的用法。

用法:

rename(*names, **rename_map)

重命名 self 的維度名稱。

主要有兩個用途：

self.rename(**rename_map) 返回張量視圖，該視圖已按照映射 rename_map 中的指定重命名了暗淡。

self.rename(*names) 返回張量視圖，使用 names 按位置重命名所有維度。使用 self.rename(None) 在張量上刪除名稱。

不能同時指定位置參數 names 和關鍵字參數 rename_map 。

例子：

>>> imgs = torch.rand(2, 3, 5, 7, names=('N', 'C', 'H', 'W'))
>>> renamed_imgs = imgs.rename(N='batch', C='channels')
>>> renamed_imgs.names
('batch', 'channels', 'H', 'W')

>>> renamed_imgs = imgs.rename(None)
>>> renamed_imgs.names
(None,)

>>> renamed_imgs = imgs.rename('batch', 'channel', 'height', 'width')
>>> renamed_imgs.names
('batch', 'channel', 'height', 'width')

警告

命名張量 API 是實驗性的，可能會發生變化。