MLMultiArray를 연결(concatenate)하는 방법
요약
PyTorch의 torch.cat 기능을 CoreML 환경의 Swift에서 구현하는 방법을 설명합니다. MLMultiArray의 concatenate 초기화 메서드를 사용하여 두 텐서를 특정 축을 따라 연결할 수 있습니다.
핵심 포인트
- PyTorch의 torch.cat 기능을 Swift에서 대체 가능
- MLMultiArray의 concatenate 초기화 메서드 활용
- CoreML 모델 입력 전처리 시 유용함
torch.cat을 대체
기계 학습 모델의 입력 전처리 과정에서 두 개의 텐서(tensor)를 임의의 축(axis)을 따라 연결하는 경우가 있습니다.
import torch
a = torch.ones(1,3,512,512)
b = torch.ones(1,1,512,512)
...
CoreML로 모델을 사용할 때 이를 Swift에서 대체하고 싶습니다.
MLMultiArray의 concatenate로 가능
CoreML의 MLMultiArray에는 concatenate 초기화가 있습니다.
let a = try MLMultiArray(shape: [1, 3, 512, 512], dataType: .int32)
let b = try MLMultiArray(shape: [1, 1, 512, 512], dataType: .int32)
let c = MLMultiArray(concatenating: [a, b],
...
🐣
프리랜서 엔지니어입니다.
업무 상담은 여기로
rockyshikoku@gmail.com
기계 학습, AR 앱(Web/iOS)을 만들고 있습니다.
기계 학습/AR 관련 정보를 발신하고 있습니다.
Discussion
AI 자동 생성 콘텐츠
본 콘텐츠는 Zenn ML의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기0