我有一个 4D 张量,我想获得最后两个维度的 argmax。torch.argmax只接受整数作为“dim”参数,而不接受元组。
我怎样才能做到这一点?
这就是我的想法,但我不知道如何匹配我的两个“索引”张量的维度。original_array是形状 [1, 512, 37, 59]。
max_vals, indices_r = torch.max(original_array, dim=2)
max_vals, indices_c = torch.max(max_vals, dim=2)
indices = torch.hstack((indices_r, indices_c))
Run Code Online (Sandbox Code Playgroud) 我是 macOS 新手,在使用 libjpeg 库时遇到问题。
使用brew install jpeg安装后,我什至尝试重新安装。
joshbone@Joshs-Mac-mini ~ % brew reinstall jpeg
==> Downloading https://homebrew.bintray.com/bottles/jpeg-9d.arm64_big_sur.bottl
Already downloaded: /Users/joshbone/Library/Caches/Homebrew/downloads/12e480d7337641afc99230ff2b626197e3206ea378eef8ccbb8b11f587afe7f4--jpeg-9d.arm64_big_sur.bottle.tar.gz
==> Reinstalling jpeg
==> Pouring jpeg-9d.arm64_big_sur.bottle.tar.gz
/opt/homebrew/Cellar/jpeg/9d: 21 files, 1001.2KB
Run Code Online (Sandbox Code Playgroud)
然后我尝试查看是否可以找到该库,我在终端中键入以下命令:“gcc -ljpeg”。这给出了以下输出:
ld: library not found for -ljpeg
clang: error: linker command failed with exit code 1 (use -v to see invocation)
Run Code Online (Sandbox Code Playgroud)
为什么找不到图书馆?我可以做什么来修复安装?我是否正确使用了这些终端命令?
编辑:我设法在 /opt/homebrew/Cellar/jpeg/9d/lib 下找到符号链接(.dylib)。但是当我运行 gcc -ljpeg -L /opt/homebrew/Cellar/jpeg/9d/lib/ 时,出现以下错误消息:
Undefined symbols for architecture arm64:
"_main", referenced from:
implicit entry/start for main executable
ld: symbol(s) not found for architecture …Run Code Online (Sandbox Code Playgroud)