NotificationsYou must be signed in to change notification settings
Fork4.4k
Star19k

Commitdb4c207

authored

Fix visual observation tensor indexing for Unity inference. (#6239)

This change corrects the tensor indexing calculation in TensorExtensions.Index()to properly support CHW (channels-height-width) format used by both Unity'sobservation writers and ONNX models during inference.

1 parenta83b3b8 commitdb4c207Copy full SHA for db4c207

File tree

1 file changed

-3

lines changed

com.unity.ml-agents/Runtime/Inference
- TensorExtensions.cs

1 file changed

-3

lines changed

`‎com.unity.ml-agents/Runtime/Inference/TensorExtensions.cs‎`

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -57,9 +57,9 @@ public static int Index(this TensorShape shape, int n, int c, int h, int w)`
`57`	`57`	`{`
`58`	`58`	`intindex=`
`59`	`59`	`nshape.Height()shape.Width()*shape.Channels()+`
`60`		`-hshape.Width()shape.Channels()+`
`61`		`-w*shape.Channels()+`
`62`		`-c;`
	`60`	`+cshape.Height()shape.Width()+`
	`61`	`+h*shape.Width()+`
	`62`	`+w;`
`63`	`63`	`returnindex;`
`64`	`64`	`}`
`65`	`65`	`}`

0 commit comments

Comments

(0)

Movatterモバイル変換

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commitdb4c207

File tree

1 file changed

1 file changed

`‎com.unity.ml-agents/Runtime/Inference/TensorExtensions.cs‎`

0 commit comments